Google Interview Problem. LEVEL: MEDIUM

Question

The problem is longer than allowed number of characters here. Read the comment in my code to understand it.
https://code.sololearn.com/c0MzlN13N6zj/?ref=app

Answer 1

i would not do a one liner in a job itw but since i'm now the one giving itws.... https://code.sololearn.com/cJfPMJjjR3uY/?ref=app