Детские задачки, сложные взрослые решения в питоне
Это решение для демонстрационных нужд. Оно специально осложнено всевозможными обощениями и графиками. Если вы решаете на скорость надо делать всё наиболее коротким путём. Этот способ для объяснений и обсуждений. Как есть.