View source for 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 = int(input())
    if numar < 1 or numar > 100000:
        print("Datele nu sunt comform restricțiilor impuse.")
        exit()
    vector = list(map(int, input().split()))
    for x in vector:
        if x > 1000000000 or x < 1:
            print("Datele nu sunt comform restricțiilor impuse.")
            exit()
    print("Datele sunt corecte.")
    return vector, numar


if __name__ == '__main__':
    vector, numar = conform_restrictiilor()
    numere(vector, numar)






</syntaxhighlight>