Efficient way?

Question

So I wrote this code and it works absolutely fine the thing is I think the code.is too big for such a problem. Can anyone suggest/write this code in a more efficient way.

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

Answer 1

This is already simple. Remember, efficient doesn't mean less code. 😃