1420 - Happy New Year

Sursa: [1]

Cerința

Se dau n numere naturale divizibile cu 9. Aceste numere se scriu unul lângă altul astfel încât să formeze un singur număr. Se calculează suma cifrelor numărului obţinut, apoi suma cifrelor sumei anterioare şi se continuă procedeul până se obţine ca rezultat un număr format dintr-o singură cifră. Cifra obţinută se înmulţeşte cu 7 şi se obţine un număr m. Se cere aflarea celui de-al m lea număr triunghiular.

Date de intrare

Programul citește de la tastatură numărul n, iar apoi n numere naturale, separate prin spații.

Date de ieșire

Programul va afișa pe ecran numărul A, reprezentând al m-lea număr triunghiular.

Restricții și precizări

1 ≤ n ≤ 1.000.000 cele n numere citite vor fi mai mici decât 2.000.000.000 un număr natural nenul se numeşte triunghiular dacă este egal cu numărul elementelor dintr-o matrice pătratică, situate deasupra diagonalei principale.

Exemplu:

Intrare

3 2016 2016 2016 Ieșire

2016

Explicație

Happy New Year !

Încărcare soluție

def calc_cifra_finala(numar):
    while numar > 9:
        suma_cifre = 0
        for cifra in str(numar):
            suma_cifre += int(cifra)
        numar = suma_cifre
    return numar

n = int(input())
numere = input().split()

numar_m = int(calc_cifra_finala(int("".join(numere))) * 7)

suma = 0
for i in range(1, numar_m + 1):
    suma += i
    if suma >= numar_m:
        print(i * (i + 1) // 2)
        break