3616 - Adfgx

Cifrul ADFGX este un cifru care folosește o matrice pătratică 5x5 și o cheie. În matricea pătratică, liniile și coloanele vor fi numerotate cu litere. Avem această matrice ca exemplu.

A D F G X

A a b c d e

D f g h i j

F k l m n o

G p q r s t

X u v w x y

Fiecărei litere îi va corespunde un cod format din linie și coloană. Spre exemplu pentru litera t din matricea de mai sus codul este GX. Literele din fiecare cod se așează sub fiecare literă din cheie. Spre exemplu pentru cheia mar și mesajul atac, folosind matricea de mai sus vom avea:

m a r

A A G

X A A

A F

După această etapă se sortează literele din cheie crescător și obținem.

a m r

A A G => AAF AXA GA

A X A

F A

Acela este mesajul codificat.

Cerința

Se citeste un text. Sa se afiseze textul codificat folosind cifrul ADFGX, daca numarul cerintei este 1, iar daca numarul cerinta este 2 se va afisa textul decodificat.

Date de intrare

Pe prima linie a fisierului de intrare adfgxin.txt se va afla un numar, reprezentand cerinta. Pe a doua linie se va afla o matrice 5 x 5. Pe a treia linie se afla cheia folosita pentru codificare. Pe a patra linie se va afisa mesajul ce trebuie codificat/decodificat,in functie de cerinta.

Date de ieșire

Se va afisa mesajul codificat/decodificat.

Restricții și precizări

• lungimea mesajului ≤ 100000
• textul va contine litere mici ale alfabetului englez
• Pentru cerinta 2, mesajul ce trebuie decodificat va contine litere mari. Pentru cerinta 1 mesajul codificat va fi afisat tot cu litere mari

Exemplul 1

adfgxin.txt

1

abcde

fghij

klmno

pqrst

uvwxy

pbinfo

romaniiatacacetatea

adfgxout.txt

Datele introduse corespund restricțiilor impuse.

FAGAFAA FDGAGA FFAAAG XGAFXX FGXAXX GADAAAA

Exemplul 2

adfgxin.txt

1

abcde

fghij

klmno

pqrst

uvwxyz

pbinfo

romaniiatacacetatea1

adfgxout.txt

Datele introduse nu corespund restricțiilor impuse.

Rezolvare

# 3616 - Adfgx
def validare(matrice_validare, mesaj_validare):           # functia de validare a datelor de intrare
    if len(mesaj_validare) > 100000:
        raise ValueError

    for linie in matrice_validare:
        if len(linie) != 5:    # fiecare linie trebuie sa aiba 5 caractere
            raise ValueError

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


def codificare(matrice_codificare, mesaj_codificare):                     # functia de rezolvare
    codificat = ""
    for litera in mesaj_codificare:
        for i in range(5):
            for j in range(5):
                if matrice_codificare[i][j] == litera:
                    codificat += "ADFGX"[i] + "ADFGX"[j]
    return codificat


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

    try:
        cerinta = int(fisier_intrare.readline())
        matrice = [list(fisier_intrare.readline().strip()) for _ in range(5)]
        cheie = fisier_intrare.readline().strip()
        mesaj = fisier_intrare.readline().strip()

        validare(matrice, mesaj)                 # apelul functiei de validare
        if cerinta == 1:
            fisier_iesire.write(codificare(matrice, mesaj))               # apelul functiei de rezolvare

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