3437 - Datorii 1

Într-o țară îndepărtată, economia este în criză. Cea mai mare problemă este lipsa de capital care creează blocaje financiare. De exemplu, o firmă X poate avea datorii către o firmă Y pe care nu le poate plăti, deoarece o altă firmă Z are datorii către firma X pe care nu le-a plătit, ş.a.m.d. Există o listă cu toate datoriile firmelor sub forma următoare: X > Y S cu semnificaţia “firma X datorează firmei Y' suma S”. Este posibil ca X să aibă mai multe datorii la firma Y (în funcţie de contractele derulate împreună) sau chiar ca X să aibă datorii la Y și Y să aibă datorii la X.

Cerințe

Cunoscând lista cu datoriile firmelor, scrieți un program care să rezolve următoarele cerințe: 1. determină numărul de firme distincte care apar în această listă; 2. realizează o situație financiară a firmelor distincte din această listă, scrise în ordine lexicografică; pentru fiecare firmă se vor determina două valori SD SP, unde SD reprezintă suma totală a datoriilor pe care firma le are către alte firme, iar SP este totalul sumelor pe care firma trebuie să le primească de la alte firme.

Date de intrare

Fișierul de intrare datorii1in.txt conține pe prima linie un număr natural C reprezentând cerința care trebuie să fie rezolvată (1 sau 2). Pe a doua linie se află un număr natural D care reprezintă numărul de înregistrări existente în lista datoriilor firmelor. Pe următoarele D linii sunt descrise datoriile firmelor, în forma specificată în enunț, câte o datorie pe o linie.

Date de ieșire

Fișierul de ieșire datoriiout.txt va conține răspunsul la cerinţa C specificată în fișierul de intrare. Dacă C=1 fișierul va conține un număr natural, reprezentând numărul de firme distincte care apar în lista menționată. Dacă C=2 fișierul va conține pentru fiecare dintre firmele distincte din lista menționată câte un singur triplet de forma X SD SP, unde X este numele firmei, iar SD și SP au semnificația din enunț pentru firma X; tripletele vor fi scrise astfel încât numele firmelor să apară în ordine lexicografică, fiecare triplet pe câte o linie a fișierului, iar X, SD și SP vor fi separate prin câte un singur spațiu.

Restricții și precizări

• Există în total cel mult 6000 de firme distincte în lista menționată de datorii.
• Numele unei firme este format din maximum 20 de caractere (litere mari şi mici ale alfabetului englez, cifre, spaţii); se face distincţie între literele mari şi literele mici în numele firmelor; nu există alte restricţii referitoare la numele firmelor.
• Două firme distincte au nume distincte. O firmă nu poate avea datorii la ea însăși.
• În descrierea unei datorii (X > Y S) există un singur spaţiu între X și >, un singur spațiu între > și Y, respectiv un singur spațiu între Y și S.
• 1 ≤ D ≤ 80000
• Sumele datorate de firme sunt numere naturale nenule ≤106.
• Dacă X și Y sunt numele a două firme distincte, iar k (k≥0) este valoarea maximă cu proprietatea că secvența formată din primele k caractere din X este identică cu secvența formată din primele caractere din Y, spunem că X precedă din punct de vedere lexicografic pe Y dacă X are doar k caractere sau dacă al k+1-lea caracter din X este mai mic decât al k+1-lea caracter din Y.
• Pentru teste valorând 30 de puncte cerinţa este 1. Pentru teste valorând 60 de puncte cerinţa este 2.
• Pentru teste valorând 40 de puncte D≤1000. Pentru teste valorând 45 de puncte numele firmelor nu

conțin spaţii.

• În concurs s-au acordat 10 puncte din oficiu. Pe site se acordă 10 puncte pentru exemple.

Exemplul 1

datoriiin.txt

1

4

Vasile Inc > Anatolia 100

ana > Anatolia 10

ana > Vasilescu Inc 5

Popa25 PF > Anatolia 30

datoriiout.txt

Datele introduse corespund restricțiilor impuse.

5

Exemplul 2

datoriiin.txt

2

5

Vasile Inc > Anatolia 100

ana > Anatolia 10

ana > Vasilescu Inc 5

Popa25 PF > Anatolia 30

Popa25 PF > ana 50

datoriiout.txt

Datele introduse corespund restricțiilor impuse.

Anatolia 0 140

Popa25 PF 80 0

Vasile Inc 100 0

Vasilescu Inc 0 5

ana 15 50

Exemplul 3

datoriiin.txt

2

6

Vasile Inc > Anatolia 100

ana > Anatolia 10

ana > Vasilescu Inc 5

Popa25 PF > Anatolia 30

Popa25 PF > ana 50

Vasile Inc > Vasile Inc 20

datoriiout.txt

Datele introduse nu corespund restricțiilor impuse.

Rezolvare

<syntaxhighlight lang="python" line="1">

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>