Editing 3437 - Datorii 1 (section)

== Rezolvare ==
<syntaxhighlight lang="python" line="1">
# 3437 - Datorii 1
from collections import defaultdict


def validare(datorii_val1):           # functia de validare a datelor de intrare
    if len(datorii_val1) > 6000:
        raise ValueError

    for X_val1, Y_val1, S_val1 in datorii_val1:
        if len(X_val1) > 20 or len(Y_val1) > 20 or S_val1 <= 0 or S_val1 > 10**6:
            raise ValueError

    file_out.write("Datele de intrare corespund restrictiilor impuse\n")


def rezolvare(datorii_val1):                     # functia de rezolvare
    firme = defaultdict(lambda: [0, 0])  # dictionar cu valori default [0, 0]
    for X_val1, Y_val1, S_val1 in datorii_val1:
        firme[X_val1][0] += S_val1  # adaugam la suma datoriilor firmei X
        firme[Y_val1][1] += S_val1  # adaugam la suma pe care firma Y trebuie sa o primeasca

    for firma in sorted(firme.keys()):  # sortam firmele lexicografic
        file_out.write(f"{firma} {firme[firma][0]} {firme[firma][1]}\n")


if __name__ == '__main__':
    file_in = open("datoriiin.txt", "r")         # declararea fisierelor
    file_out = open("datoriiout.txt", "w")       # fisierul out trebuie declarat cu optiunea "w" (write)

    # din cauza datelor de intrare pot aparea 2 tipuri de erori, valueError sau IndexError pe care le tratam
    try:
        C = int(file_in.readline())      # citirea cerintei
        D = int(file_in.readline())      # citirea numarului de datorii
        datorii_val = []
        for _ in range(D):
            X_val, _, Y_val, S_val = file_in.readline().split()
            S_val = int(S_val)
            datorii_val.append((X_val, Y_val, S_val))

        validare(datorii_val)                 # apelul functiei de validare
        if C == 2:
            rezolvare(datorii_val)             # apelul functiei de rezolvare

    except ValueError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")
    except IndexError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")

</syntaxhighlight>