Editing 1481 - Gradina Piticului

==  Enunț ==
Un pitic pasionat de numere trebuie să-și pună flori în grădină. El are de plantat m rânduri cu flori, aceeași floare pe tot rândul. Rândurile sunt numerotate de la 1 la m. Având la dispoziție suficiente specii de flori, piticul nostru s-a gândit să le planteze folosind următorul algoritm matematic: pe rândurile care sunt numere prime, va planta exact floarea numerotată cu numărul prim respectiv, iar pe celelalte rânduri va planta floarea numerotată cu suma divizorilor primi ai numărului neprim.
== Cerinţa ==
Să se realizeze un program care să afişeze ordinea de așezare a florilor pe cele '''m''' rânduri.
== Date de intrare ==
Fișierul de intrare '''pitic.in''' conține pe prima linie numărul '''m'''.
== Date de ieşire ==
Fișierul de ieșire '''pitic.out''' va conține pe prima un șir de '''m''' numere, separate prin câte un spațiu, fiecare număr reprezentând specia de floare ce urmează să fie plantată pe rândul respectiv.
== Restricții și precizări ==
* 1 &les; m &les; 1000
== Exemplu ==
; pitic.in
: 6
; pitic.out
: Datele introduse corespund restricțiilor impuse.
: 0 2 3 2 5 5
== Rezolvare ==
<syntaxhighlight lang="python" line>
import math


def prim(n):
    d = 2
    cnt = 1
    while n > 1:
        p = 0
        while n % d == 0:
            p += 1
            n //= d
        if p > 0:
            cnt *= (p + 1)
        d += 1
        if d * d > n:
            d = n
    if cnt == 2:
        return 1
    else:
        return 0


def sumprim(n):
    s = 0
    d = 2
    while n > 1:
        p = 0
        while n % d == 0:
            p += 1
            n //= d
        if p > 0:
            s += d
        d += 1
        if d * d > n:
            d = n
    return s


def validare_date(m):
    return 1 <= m <= 1000

def main():
    with open("pitic.in") as fin, open("pitic.out", "w") as fout:
        m = int(fin.readline().strip())
        if validare_date(m):
            fout.write("Datele de intrare corespund restrictiilor impuse.\n")
            for i in range(1, m + 1):
                if prim(i):
                    fout.write(str(i) + " ")
                else:
                    fout.write(str(sumprim(i)) + " ")
        else:
            fout.write("Datele de intrare nu corespund restrictiilor impuse.")



</syntaxhighlight>