0499 - Numarare5

Sursa: - 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 ⩽ n ⩽ 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:
51 15
51 132
842 77
15 132

Exemplul 2

Intrare

5

10 2 -3 4 5

Ieșire

Datele nu corespund restricțiilor impuse.

Rezolvare

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

Explicație rezolvare

   Programul de mai sus face:
-> 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.
-> 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().
-> 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.
   Î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ă.