4163 - Seif

De la Universitas MediaWiki

Cerința

Seiful SEPI, în care sunt depozitate premiile olimpiadelor de informatică, este securizat cu un cifru sub forma unei matrice A de formă pătratică cu N linii și N coloane, unde N este un număr natural par. Liniile și coloanele sunt numerotate începând cu 1.

Matricea-cifru A este formată din N / 2 chenare. Al i-lea chenar (1 ≤ i ≤ N / 2) va conține elementele aflate pe marginea matricei A, după excluderea primelor i - 1 chenare. Ordinea elementelor acestui chenar este obținută pornind din poziția (i, i), parcurgând latura de sus de la stânga la dreapta, latura din dreapta de sus în jos, latura de jos de la dreapta la stânga și apoi latura din stânga de jos în sus. Fiecare chenar acționează ca un rotor al cifrului și elementele sale pot fi permutate circular către stânga sau către dreapta cu un anumit număr de poziții. Asupra matricei-cifru se aplică o succesiune de T de operații de permutare circulară a elementelor unui chenar dat cu un număr oarecare de poziții, către stânga sau către dreapta. O operație de permutare circulară are următoarea structură: chenar pozitii sens, unde chenar reprezintă numărul de ordine al chenarului asupra căruia se aplică permutarea, pozitii reprezintă numărul de poziții cu care se permută, iar sens este un caracter: S pentru stânga sau D pentru dreapta.

Cunoscând numărul natural N, cele N x N elemente ale matricei-cifru precum și succesiunea de T operații de permutare circulară a unor chenare, să se determine configurația finală a matricei, cea care va permite deschiderea seifului!

Date de intrare

Fișierul de intrare seif.in conține pe prima linie numărul natural N. Pe fiecare dintre următoarele N linii, câte N numere naturale reprezentând matricea-cifru inițială. Pe linia N + 2 numărul natural T reprezentând numărul de operații de permutare. Pe următoarele T linii sunt scrise cele T operații de permutare circulară, câte o operație pe fiecare linie, în formatul descris în enunț: chenar pozitii sens.

Date de ieșire

Fișierul de ieșire seif.out va conține pe primele N linii câte N numere reprezentând elementele matricei după aplicarea celor T operații de permutare circulară.

Restricții și precizări

  • 2 ≤ N ≤ 500
  • N este număr par
  • 1 ≤ T ≤ 100.000
  • Numărul de poziții cu care se face o permutare este nenul și ≤ 1.000.000.
  • Elementele matricei-cifru sunt numere naturale ≤ 2.000.000.000.
  • Pe un chenar se pot face mai multe operații de permutare circulară.
  • Elementele scrise pe aceeași linie în fișierul de intrare, respectiv în fișierul de ieșire sunt separate prin câte un singur spațiu.

Exemplu 1

Intrare

2
1 2
4 3
3
1 1 D
1 3 S
1 3 D

Iesire

4 1
3 2

Rezolvare

def citire_date():
    with open("seif.in", "r") as f:
        N = int(f.readline().strip())
        matrice = [list(map(int, f.readline().strip().split())) for _ in range(N)]
        T = int(f.readline().strip())
        operatii = [f.readline().strip().split() for _ in range(T)]
        operatii = [(int(op[0]), int(op[1]), op[2]) for op in operatii]
    return N, matrice, T, operatii

def extrage_chenar(matrice, chenar, N):
    elementi = []
    start = chenar - 1
    end = N - chenar

    for j in range(start, end + 1):
        elementi.append(matrice[start][j])
    for i in range(start + 1, end + 1):
        elementi.append(matrice[i][end])
    for j in range(end - 1, start - 1, -1):
        elementi.append(matrice[end][j])
    for i in range(end - 1, start, -1):
        elementi.append(matrice[i][start])

    return elementi

def insereaza_chenar(matrice, chenar, N, elementi):
    index = 0
    start = chenar - 1
    end = N - chenar

    for j in range(start, end + 1):
        matrice[start][j] = elementi[index]
        index += 1
    for i in range(start + 1, end + 1):
        matrice[i][end] = elementi[index]
        index += 1
    for j in range(end - 1, start - 1, -1):
        matrice[end][j] = elementi[index]
        index += 1
    for i in range(end - 1, start, -1):
        matrice[i][start] = elementi[index]
        index += 1

def permutare_circulara(elementi, pozitii, sens):
    lungime = len(elementi)
    pozitii = pozitii % lungime
    if sens == 'S':
        return elementi[pozitii:] + elementi[:pozitii]
    elif sens == 'D':
        return elementi[-pozitii:] + elementi[:-pozitii]

def aplica_operatii(N, matrice, operatii):
    for chenar, pozitii, sens in operatii:
        elementi = extrage_chenar(matrice, chenar, N)
        elementi_permutati = permutare_circulara(elementi, pozitii, sens)
        insereaza_chenar(matrice, chenar, N, elementi_permutati)

def scrie_rezultate(matrice):
    with open("seif.out", "w") as f:
        for linie in matrice:
            f.write(" ".join(map(str, linie)) + "\n")

def main():
    N, matrice, T, operatii = citire_date()

    # Verificarea restrictiilor
    assert 2 <= N <= 500, "N trebuie sa fie intre 2 si 500"
    assert N % 2 == 0, "N trebuie sa fie par"
    assert 1 <= T <= 100000, "T trebuie sa fie intre 1 si 100.000"
    for op in operatii:
        assert 1 <= op[1] <= 1000000, "Numarul de pozitii trebuie sa fie intre 1 si 1.000.000"
        assert op[2] in ['S', 'D'], "Sensul trebuie sa fie 'S' sau 'D'"
    
    aplica_operatii(N, matrice, operatii)
    scrie_rezultate(matrice)

if __name__ == "__main__":
    main()