лестничные маньяки

До чего же Питон кривой - вот как приходится извращаться, чтобы уложить нетривиальный цикл в одну строчку.