Предлагаю ознакомиться с моим решением данной задачи (
https://codeforces.com/contest/1016/problem/B) по ссылке:
https://codeforces.com/contest/1016/submission/41169503
Решение получилось самым быстрым среди других решений на третьем Питоне, проверить можно тут:
https://codeforces.com/contest/1016/status/B/page/... (необходимо выбрать язык Python 3 в фильтре статуса)