Rosetta Problems

problem b

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