Условия задачи по ссылки: https://dzen.ru/a/ZQN6p4PafWxlmS0p
Решение задачи в одну строку:
Пример использования: