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 ⩽ m ⩽ 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
if __name__ == "__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>