1754 - Munți

Sursa: [1]

Cerinţa

Se dă un număr natural n. Dacă numărul este norocos afișați cele n numere consecutive care adunate dau pătratul acestuia.

Date de intrare

Programul citește de la tastatură numărul n.

Date de ieșire

Daca n este norocos, programul va afișa pe ecran cele n numere consecutive care adunate dau n * n, separate prin spații. In caz contrar, programul va afișa mesajul NU ESTE NOROCOS.

Restricţii şi precizări

• 1 ⩽ n ⩽ 1000

Exemplul 1

Intrare

7

Ieșire

4 5 6 7 8 9 10

Exemplul 2

Intrare

8

Ieșire

NU ESTE NOROCOS

Rezolvare

<syntaxhighlight lang="python" line>

1. 1754

def validate_input(n, x_list):

   if not 1 <= n <= 1000000:
       print("N trebuie sa fie intre 1 si 1000000.")
       exit()
   if not all(1 <= x <= 1000000000 for x in x_list):
       print("Inaltimea initiala trebuie sa fie intre 1 si 1000000000.")
       exit()

def main():

   with open("munti.in", "r") as f_in, open("munti.out", "w") as f_out:
       n = int(f_in.readline())
       x_list = list(map(int, f_in.readline().split()))
       validate_input(n, x_list)
       s = sum(x_list)
       if s % n == 0:
           f_out.write(str(n))
       else:
           f_out.write(str(n - 1))

main()

</syntaxhighlight>

Explicatie rezolvare

Acest program validează inputul pentru o problemă specifică și apoi calculează un rezultat pe baza inputului. În ceea ce privește validarea inputului, funcția validate_input primește doi parametri: n și x_list. n reprezintă numărul de înălțimi de verificat, iar x_list este o listă de înălțimi inițiale. În cazul în care n este mai mic decât 1 sau mai mare decât 1.000.000 sau oricare dintre înălțimi este mai mică decât 1 sau mai mare decât 1.000.000.000, funcția afișează un mesaj de eroare și apoi iese din program. Funcția main deschide două fișiere, "munti.in" și "munti.out", citind din fișierul de intrare munti.in valoarea lui n și lista de înălțimi x_list. Apoi, apelând funcția validate_input, se verifică validitatea inputului. Dacă suma tuturor înălțimilor din listă este divizibilă cu n, se afișează n în fișierul de ieșire munti.out. În caz contrar, se afișează n-1. În general, acest program se ocupă de o problemă specifică care implică validarea inputului și apoi efectuarea unor calcule pe baza acelui input.