Editing 1629 - Qmat

== Enunț ==
Spunem că două matrice a și b sunt egale dacă au același număr de linii n și același număr de coloane m și pentru orice pereche de indici i, j (1 ≤ i ≤ n, 1 ≤ j ≤ m) a[i][j] = b[i][j].

== Cerința ==
Se dau două seturi de N, respectiv Q matrice binare (cu valori 0 sau 1), pentru fiecare matrice fiind precizat numărul de linii respectiv de coloane. Să se afișeze numărul aparițiilor matricelor din al doilea set în primul.

== Date de intrare ==
Fișierul de intrare qmatin.txt conține pe prima linie numărul N, urmat de N matrice, M1, M2, … Mn. Pentru fiecare matrice Mi este precizat, pe o linie, separate printr-un spațiu, numărul de linii li, respectiv de coloane ci, iar pe următoarele li linii câte ci cifre binare separate prin câte un spațiu. După cele N matrice se află și numărul Q urmat de asemenea de Q matrice descrise la fel ca și cele anterioare.

== Date de ieșire ==
Fișierul de ieșire qmatout.txt va conține pe prima linie numărul sol, reprezentând numărul de matrice din al doilea set ce apar în primul.

== Restricții și precizări ==
* 1 &les; '''n, q''' &les; 10.000
* pentru orice matrice dată 1 ≤ nr. de linii, nr. de coloane ≤ 10

== Exemplul 1 ==
; Intrare
; qmatin.txt
: 5
: 2 6
: 0 0 0 1 1 1
: 1 0 1 1 0 0
: 7 4
: 0 0 0 1
: 0 0 0 0
: 0 1 1 0
: 0 1 1 0
: 1 1 1 1
: 0 1 0 0
: 0 1 0 1
: 5 1
: 0
: 0
: 1
: 0
: 1
: 3 5
: 0 1 1 1 1
: 1 1 1 0 0
: 0 0 1 1 0
: 9 2
: 0 1
: 1 1
: 1 1
: 1 0
: 1 1
: 0 1
: 0 1
: 0 0
: 1 1
: 3
: 3 5
: 0 1 1 1 1
: 1 1 1 0 0
: 0 0 1 1 0
: 9 2
: 0 1
: 1 1
: 1 1
: 1 0
: 1 1
: 0 1
: 0 1
: 0 0
: 1 1
: 2 3
: 1 1 1
: 1 1 1
; Ieșire
: Datele de intrare corespund restricțiilor impuse
; qmatout.txt
: 2

== Exemplul 2 ==
; Intrare
; qmatin.txt
: 5
: 2 6
: 2 0 0 1 1 1
: 1 0 1 1 0 0
: 7 4
: 0 0 0 1
: 0 0 0 0
: 0 1 1 0
: 0 1 1 0
: 1 1 1 1
: 0 1 0 0
: 0 1 0 1
: 5 1
: 0
: 0
: 1
: 0
: 1
: 3 5
: 0 1 1 1 1
: 1 1 1 0 0
: 0 0 1 1 0
: 9 2
: 0 1
: 1 1
: 1 1
: 1 0
: 1 1
: 0 1
: 0 1
: 0 0
: 1 1
: 3
: 3 5
: 0 1 1 1 1
: 1 1 1 0 0
: 0 0 1 1 0
: 9 2
: 0 1
: 1 1
: 1 1
: 1 0
: 1 1
: 0 1
: 0 1
: 0 0
: 1 1
: 2 3
: 1 1 1
: 1 1 1
; Ieșire
: Datele de intrare NU corespund restricțiilor impuse

== Rezolvare ==
<syntaxhighlight lang="python" line>
#1629 - Qmat

def este_matrice_valida(matrice):
    if len(matrice) == 0 or len(matrice[0]) == 0:
        return False

    lungime_rand = len(matrice[0])
    if any(len(rand) != lungime_rand for rand in matrice):
        return False

    if any(element not in (0, 1) for rand in matrice for element in rand):
        return False

    return True


def citeste_matrice(fisier):
    nr_randuri, nr_coloane = map(int, fisier.readline().split())
    matrice = [list(map(int, fisier.readline().split())) for _ in range(nr_randuri)]
    return matrice


def numara_aparitii(set_matrice_1, set_matrice_2):
    numar_aparitii = 0
    for matrice_2 in set_matrice_2:
        for matrice_1 in set_matrice_1:
            if matrice_1 == matrice_2:
                numar_aparitii += 1
                break
    return numar_aparitii


def main():
    with open("qmatin.txt", "r") as fisier:
        n = int(fisier.readline())
        set_matrice_1 = [citeste_matrice(fisier) for _ in range(n)]

        q = int(fisier.readline())
        set_matrice_2 = [citeste_matrice(fisier) for _ in range(q)]

    if all(este_matrice_valida(matrice) for set_matrice in [set_matrice_1, set_matrice_2] for matrice in set_matrice):
        print("Datele de intrare corespund restricțiilor impuse")
        rezultat = numara_aparitii(set_matrice_1, set_matrice_2)
        with open("qmatout.txt", "w") as fisier_out:
            fisier_out.write(str(rezultat))
    else:
        print("Datele de intrare NU corespund restricțiilor impuse")


if __name__ == "__main__":
    main()

</syntaxhighlight>