Rosetta Problems

problem a

https://www.acwing.com/problem/content/2701