View source for 0499 - Numarare5

Sursa: [https://www.pbinfo.ro/probleme/499/numarare5 - Numarare5]
----
== Cerinţa ==
Se dă un vector cu '''n''' 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 '''n''', iar apoi '''n''' 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 introduse corect."''', apoi pe un rând nou afișează pe ecran '''numărul C''', reprezentând valoarea cerută. În caz contrar, se va afișa pe ecran: '''"Datele nu corespund restricțiilor impuse."'''.

== Restricţii şi precizări ==
* 1 &les; '''n''' &les; 200
* elementele vectorului vor fi cuprinse între '''0''' și '''1.000.000.000'''

== Exemple ==
===Exemplul 1===
; Intrare
: 6
: 51 842 15 28 77 132
; Ieșire
: Datele sunt introduse corect.
: 4
===Explicație exemplul 1===
Perechile de elemente cu aceeaşi sumă cifrelor sunt:<br>
'''51 15'''<br>
'''51 132'''<br>
'''842 77'''<br>
'''15 132'''
===Exemplul 2===
; Intrare
: 5
: 10 2 -3 4 5
; Ieșire
: Datele nu corespund restricțiilor impuse.
<br>
== Rezolvare == 
<syntaxhighlight lang="python" line>
# 0499

def suma_cifrelor(n):
    suma = 0
    while n:
        suma += n % 10
        n = n//10
    return suma


def cate_perechi_cu_aceeasi_suma_cifrelor(vector, n):
    s = [suma_cifrelor(x) for x in vector]
    C = 0
    for i in range(n):
        for j in range(i+1, n):
            if s[i] == s[j]:
                C += 1
    print(C)


def citire_conform_restrictiilor(vector, n):
    if n < 1 or n > 200:
        print("Datele nu corespund restricțiilor impuse.")
        exit()
    for x in vector:
        if x < 0 or x > 1000000000:
            print("Datele nu corespund restricțiilor impuse.")
            exit()
    if n != len(vector):
        print("Datele nu corespund restricțiilor impuse.")
        exit()
    print("Datele sunt introduse corect.")


if __name__ == '__main__':
    n = int(input())
    vector = list(map(int, input().split()))
    citire_conform_restrictiilor(vector, n)
    cate_perechi_cu_aceeasi_suma_cifrelor(vector, n)












</syntaxhighlight>
== Explicație rezolvare ==
    Programul de mai sus face:<br>-> Definirea '''funcției suma_cifrelor(n)''' care primește un număr întreg n și calculează suma cifrelor acestuia. Funcția utilizează o buclă while care rulează până când n devine zero, la fiecare iterație extrăgând ultima cifră a lui n folosind operatorul modulo % și adăugând-o la suma suma. După aceea, n este împărțit la 10 folosind operatorul de divizare întreagă // pentru a se trece la următoarea cifră. Suma cifrelor este returnată la final.<br>-> Definirea '''funcției cate_perechi_cu_aceeasi_suma_cifrelor(vector, n)''' care primește un vector de numere întregi vector și un număr întreg n reprezentând lungimea vectorului. Funcția calculează suma cifrelor pentru fiecare element din vector utilizând funcția suma_cifrelor() și stochează rezultatele într-un alt vector s. Apoi, folosește două bucle for pentru a compara suma cifrelor a fiecărui element din vectorul s cu suma cifrelor a celorlalte elemente din vector și numără câte perechi au aceeași sumă a cifrelor. Rezultatul este afișat la final cu ajutorul funcției print().<br>-> Definirea '''funcției citire_conform_restrictiilor(vector, n)''' care primește un vector vector și un număr n reprezentând lungimea vectorului și verifică dacă acestea respectă anumite restricții. Restricțiile impuse includ: n trebuie să fie între 1 și 200, fiecare element din vector trebuie să fie între 0 și 1000000000, iar lungimea vectorului vector trebuie să fie egală cu n. Dacă oricare dintre restricții nu este respectată, se afișează "Datele nu corespund restricțiilor impuse." și programul se încheie cu ajutorul funcției exit(). Dacă toate restricțiile sunt respectate, se afișează "Datele sunt introduse corect." și se continuă programul.<br>   În '''secțiunea if __name__ == '__main__':''' se citește de la tastatură numărul n și vectorul vector folosind funcția input() și list(map(int, input().split())). Apoi, se apelează funcția citire_conform_restrictiilor(vector, n) pentru a verifica dacă datele introduse respectă restricțiile impuse. În final, se apelează funcția cate_perechi_cu_aceeasi_suma_cifrelor(vector, n) pentru a afișa valoarea cerută în problemă.