Odpowiedź
Program w załączniku. Spacji nie usuwałam, bo uznałam, że nie chodzi o zmniejszenie czytelności tylko o oszczędności typu sprawdzanie < 1 zamiast sprawdzania == 0 co daje jeden znak mniej... :-)
Wyjaśnienie
Program napisany i przetestowany przy użyciu Colab.