I have a waterproof notebook for situations just like these, though it got mostly used to sketch out public talks - during showers I would often think about the points I wanted to make, and suddenly, new insights would dawn on me.