Самое смешное это преставка Python в конце. Это когда вот прямо таки НУ СОВСЕМ ЛЕНЬ. Проявите хоть немного самостоятельности, либо напишите алгоритм в псевдкоде и Вам его в питоне начеркают, либо гуглите решения на другом языке и перепишите под питон, Вы хотите что бы за Вас решили и МАТЕМАТИКУ и ПРОГРАММИРОВАНИЕ, верх наглости.