"if" in "case"

Question

Подскажите, пожалуйста, как исправить программу, чтобы она работала потому же принципу
https://code.sololearn.com/c85JtDTAOgzW/?ref=app

Answer 1

Translated:
Please tell me how to fix the program so that it works because it works because of the same principle