Оптимизировать код динамической задачи на python
Есть задача https://pastebin.com/R7JJkee2 и решение на Python, но не проходит по времени (ограничение в 1 секунду), необходимо доработать код.