maybe this would happen...
naruto kills pain
naruto gets exhausted from his battle with the six pains
tobi / madara shows up, trying to kill him...
sasuke arrives in konoha seeing it destroyed and seeing naruto having a hard time, beat up and madara trying to kill him...
all memories comes...