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

#2669 Count5

def numere(vector, numar):
    suma = list()
    for i in range(numar):
        y=vector[i]
        s=0
        while y!=0:
            s+=y%10
            y//=10
        suma.append(s)
    print(suma)
    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)