Editing 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 ==
<syntaxhighlight lang="python" line="1">
# 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.")

</syntaxhighlight>