Это уже готовые решения на Python

Рекомендую всё же решать самим, так больше чему научитесь, чем копированию.
Впрочем это ваше дело.