Editing 1323 - Matrice Rara (section)

== Rezolvare ==
<syntaxhighlight lang="python" line="1">
# 1323 - Matrice Rara
def validare(matrice_a1, matrice_b1):           # functia de validare a datelor de intrare
    for matrice in [matrice_a1, matrice_b1]:
        if len(matrice) > 1000000:
            raise ValueError

        for element in matrice:
            if element[2] < -1000000000 or element[2] > 1000000000:
                raise ValueError

    fisier_iesire.write("Datele introduse corespund restrictiilor impuse.\n")


def suma_matrici(matrice_a1, matrice_b1):                     # functia de rezolvare
    suma = {}
    for element in matrice_a1 + matrice_b1:
        if (element[0], element[1]) not in suma:
            suma[(element[0], element[1])] = 0
        suma[(element[0], element[1])] += element[2]

    matrice_suma1 = [(k[0], k[1], v) for k, v in suma.items() if v != 0]
    matrice_suma1.sort()

    return matrice_suma1


if __name__ == '__main__':
    fisier_intrare = open("matrice_rarain.txt", "r")         # declararea fisierelor
    fisier_iesire = open("matrice_raraout.txt", "w")       # fisierul out trebuie declarat cu optiunea "w" (write)

    try:
        n, m, N1, N2 = map(int, fisier_intrare.readline().split())
        matrice_a = [list(map(int, fisier_intrare.readline().split())) for _ in range(N1)]
        matrice_b = [list(map(int, fisier_intrare.readline().split())) for _ in range(N2)]

        validare(matrice_a, matrice_b)                 # apelul functiei de validare

        matrice_suma = suma_matrici(matrice_a, matrice_b)               # apelul functiei de rezolvare
        fisier_iesire.write(f"{len(matrice_suma)}\n")
        for elem in matrice_suma:
            fisier_iesire.write(f"{elem[0]} {elem[1]} {elem[2]}\n")

    except ValueError:
        fisier_iesire.write("Datele introduse nu corespund restrictiilor impuse.")
    except IndexError:
        fisier_iesire.write("Datele introduse nu corespund restrictiilor impuse.")

</syntaxhighlight>