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):
        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)

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.