Plzz help me i don't know much about python and be sure code will be like that but check the bug and correct it plzzz soon as

Question

https://code.sololearn.com/cntmnPcQl5jg/?ref=app

Answer 1

https://code.sololearn.com/cwZrsgfDUkQz/?ref=app
here you go