`sum+1` это такая бесконечность?

Вообще, с учётом ограничений на задачу, можно в начале сказать

inf = 10**9 и использовать inf вместо sum -- вроде должно чуть упросить код, и название переменной будет более самодокументирующиеся