Оптимизация решения: как убрать ветку else и использовать for/else?