Is there any easiest method to solve this problem, my code is bit bigger

Question

If you make it a little easier then it will be good for me.
https://code.sololearn.com/ch726EEKrRAw/?ref=app