Editing 2669 - Count5

Sursa: [https://www.pbinfo.ro/probleme/2669/count5 - 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 &les; '''numar''' &les; 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
<br>
== 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.