2669 - Count5
Sursa: - Count5
Cerinţa
Se dă un vector cu numar numere naturale. Să se determine câte dintre perechile de elemente din vector sunt formate din valori cu aceeași sumă a cifrelor.
Date de intrare
Programul citește de la tastatură numărul numar, iar apoi numar numere naturale, separate prin spaţii, reprezentând elementele vectorului.
Date de ieșire
Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele sunt corecte.", iar apoi se afișează pe ecran numărul C, reprezentând valoarea cerută. În caz contrar, se va afișa pe ecran: "Datele nu sunt comform restricțiilor impuse.".
Restricţii şi precizări
- 1 ⩽ numar ⩽ 100.000
- cele numar numere citite vor fi mai mici decât 1.000.000.000
Exemple
Exemplul 1
- Intrare
- 6
- 51 842 15 28 77 132
- Ieșire
- Datele sunt corecte.
- 4
Exemplul 2
- Intrare
- 10
- 100 10000 100 20 1 20 30 10 98 999999999999999
- Ieșire
- Datele nu corespund restricțiilor impuse.
Exemplul 3
- Intrare
- 5
- 1 11 111 1111 11111
- Ieșire
- Datele sunt corecte.
- 0
Rezolvare
<syntaxhighlight lang="python" line>
- 2669 Count5
def numere(vector, numar):
suma = list() for i in range(numar): element = vector[i] suma_cifre = 0 while element != 0: suma_cifre += element % 10 element //= 10 suma.append(suma_cifre) C = 0 for i in range(numar): for j in range(i + 1, numar): if suma[i] == suma[j]: C += 1 print(C)
def conform_restrictiilor(numar, vector):
if numar < 1 or numar > 100000: print("Datele nu sunt comform restricțiilor impuse.") return False for x in vector: if x > 1000000000 or x < 1: print("Datele nu sunt comform restricțiilor impuse.") return False print("Datele sunt corecte.") return True
if __name__ == '__main__':
numar = int(input()) vector = list(map(int, input().split())) if conform_restrictiilor(numar, vector) is True: numere(vector, numar)
</syntaxhighlight>
Explicaţie cod
Acest cod are ca scop determinarea numarului de perechi de numere dintr-un vector care au aceeasi suma a cifrelor.
La inceput, functia numere(vector, numar) calculeaza suma cifrelor fiecarui element din vector si salveaza valorile in lista suma. Apoi, se parcurge lista suma si se compara fiecare element cu toate elementele din dreapta sa. Daca doua elemente au aceeasi valoare, se incrementeaza un contor C. La final, functia afiseaza valoarea contorului.
Functia conform_restrictiilor(numar, vector) verifica daca datele de intrare respecta restrictiile impuse. Aceasta verifica daca numarul de elemente din vector este cuprins intre 1 si 100000, si daca fiecare element din vector este cuprins intre 1 si 1000000000. Daca datele sunt in regula, functia returneaza True, altfel returneaza False.
In programul principal, se citeste numarul de elemente din vector si vectorul de la intrare, iar apoi se verifica daca datele respecta restrictiile. Daca datele sunt in regula, se apeleaza functia numere(vector, numar) pentru a calcula numarul de perechi cu aceeasi suma a cifrelor.