This is if the Night King just had to die.
I would have liked some sort of a fight between the generals and those with Valerian Steel. Have them fight so we can see the skill of the White Walkers and heroes, and have a hero or two go down and all of the generals. As each general goes down his portion of Wights fall as well.
The Dothraki, Unsullied, and other minor characters are fighting Wights until they are overwhelmed and pushed back into Winterfell, just as it happened in the show, but with each general falling they start to gain ground.
Dragon fight still happens, but Dany goes to the battlefield to help while Jon goes to Bran after they knock the Night King off of his dragon. Still have the dragonfire on the Night King scene. He goes on to meet Bran but he's by himself.
When he gets there Reek goes after him and dies like he did. Jon gets there just as Reek goes down.
Jon and Night King have an epic fight. The Night King gets the best of Jon and as he's about to strike the death blow on Bran, Arya comes flying in like she did killing the Night King to Jon's amazement (this is if Arya just had to be the killer).