2900 - Joc 9

Enunț

Amalia a adunat n cartonașe pe care le poate folosi în mai multe jocuri. Pe fiecare cartonaș este scris un singur număr care are cel puțin două cifre și cel mult nouă cifre. La jocul Numersum, ea utilizează cartonașele ce au numere pentru care prima cifră este egală cu suma celorlalte cifre din număr. Amalia dorește să aleagă cartonașele pentru acest joc și să le așeze în ordine crescătoare, după numerele scrise pe ele.

Cerința

Scrieți un program care, cunoscând numărul de cartonașe n și numerele scrise pe cele n cartonașe, determină în ordine crescătoare numerele pe care Amalia le poate utiliza la jocul Numersum.

Date de intrare

Fișierul de intrare jocin.txt conține pe prima linie numărul natural n ce reprezintă numărul de cartonașe. Pe a doua linie din fișier se află n numere naturale c1, c2, … cn, reprezentând numerele scrise pe cele n cartonașe.

Date de ieșire

Fișierul de ieșire jocout.txt conține o singură linie pe care se află în ordine crescătoare numerele scrise pe cartonașele ce se pot utiliza la jocul Numersum, separate prin câte un spațiu.

Restricții și precizări

• 2 ⩽ n ⩽ 40.000
• 10 ⩽ ci ⩽ 999.999.999
• În setul de cartonașe al Amaliei există cel puțin un cartonaș ce poate fi ales la jocul Numersum

Exemplul 1

Intrare

jocin.txt

7

211 58 55 514 101 9867 101

Ieșire

Datele de intrare corespund restricțiilor impuse

jocout.txt

55 101 101 211 514

Explicație

Numerele din șirul dat ce se pot utiliza la jocul Numersum sunt: 211 (2 = 1 + 1), 55 (5=5), 514, 101 (1 = 0 + 1), 101 și se vor afișa în ordine crescătoare.

Exemplul 2

Intrare

jocin.txt

2

9 511

Ieșire

Datele de intrare NU corespund restricțiilor impuse

Rezolvare

#2900 - Joc9

def validare_date(n, cartonase):
    if not (3 <= n <= 40000):
        return False
    for numar in cartonase:
        if not (10 <= numar < 1000000000):
            return False
    return True


def numere_pentru_joc(cartonase):
    rezultat = []
    for numar in cartonase:
        cifre = [int(digit) for digit in str(numar)]
        if cifre[0] == sum(cifre[1:]):
            rezultat.append(numar)
    return sorted(rezultat)


def main():
    with open("jocin.txt", "r") as f:
        n = int(f.readline().strip())
        cartonase = list(map(int, f.readline().split()))

    if validare_date(n, cartonase):
        print("Datele de intrare corespund restricțiilor impuse")
    else:
        print("Datele de intrare NU corespund restricțiilor impuse")
        exit(0)

    rezultat = numere_pentru_joc(cartonase)

    with open("jocout.txt", "w") as f:
        f.write(" ".join(map(str, rezultat)))


if __name__ == "__main__":
    main()