In round #643 Div 2 , for question B, I did this solution using PyPy 3: https://codeforces.net/contest/1355/submission/80386227

it passed the protests but TLE'd test case #26 after system testing I did the same code in CPP , and it ran flawlessly: https://codeforces.net/contest/1355/submission/80387439

Is there any tip for me, who codes in python. relatively new to cpp. Should there be time extension for python as in other platforms? or is there something I can do to my code to make it faster?