Whenever that sort of thing happens to me in school, I quickly jot down the little thoughts on the edge of my paper, then I find I'll stop thinking about it, and if they come back later, I'll do it again.
If I really need to think it through ill go over everything i wrote down. But this doesnt always work.