Ето едно просто решение на задачата, което минава със 100/100 в judge: https://pastebin.com/dF5Tt6g7