View source for 3856 - Scytale

== Scytale ==
Cifrul '''Scytale''' functioneaza asa:
se da un mesaj cu '''n''' litere si un numar '''m''';
mesajul trebuie asezat intr-o matrice cu '''m''' linii si n coloane;
pentru mesajul '''WEAREDISCOVEREDFLEEATONCE''' cu '''25''' de litere si numarul '''m=3''', mesajul va fi pozitionat in matrice asa:

'''W . . E . . A . . R . . E . . D . . I . . S . . C'''

'''. O . . V . . E . . R . . E . . D . . F . . L . .'''

'''. . E . . E . . A . . T . . O . . N . . C . . E .'''

== Cerința ==
Dandu-se un numar c, mesajul si numarul m:
* pentru '''c=1''', sa se afiseze mesajul criptat
* pentru '''c=2''', sa se afiseze mesajul decriptat
== Date de intrare ==
Fișierul de intrare '''scytalein.txt''' conține pe prima linie numărul '''c''', mesajul si numarul '''m'''.
== Date de ieșire ==
Fișierul de ieșire '''scytaleout.txt''' va conține pe prima linie mesajul criptat sau decriptat, in functie de '''c'''.
== Restricții și precizări ==
* '''1 ≤ n*m ≤ 1000000'''
* mesajul va contine numai litere mari
== Exemplul 1 ==
; scytalein.txt
: 1 WEAREDISCOVEREDFLEEATONCE 3
; scytaleout.txt
: Datele introduse corespund restricțiilor impuse.
: WOEEVEAEARRTEEODDNIFCSLEC
== Exemplul 2 ==
; scytalein.txt
: 2 WOEEVEAEARRTEEODDNIFCSLEC 3
; scytaleout.txt
: Datele introduse corespund restricțiilor impuse.
: WEAREDISCOVEREDFLEEATONCE
== Exemplul 3 ==
; scytalein.txt
: 2 WOEEVEAEARRTEEODDNIFCSLEC 300000
: Datele introduse nu corespund restricțiilor impuse.
== Rezolvare ==
<syntaxhighlight lang="python" line="1">
# 3856 - Scytale
def validare(mesaj_validare, m_validare):           # functia de validare a datelor de intrare
    if len(mesaj_validare) * m_validare > 1000000:
        raise ValueError

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


def criptare(mesaj_criptare, m_criptare):                     # functia de criptare
    criptat = ""
    for i in range(m_criptare):
        for j in range(i, len(mesaj_criptare), m_criptare):
            criptat += mesaj_criptare[j]
    return criptat


def decriptare(mesaj_decriptare, m_decriptare):                     # functia de decriptare
    n = len(mesaj_decriptare) // m_decriptare
    decriptat = ""
    for i in range(n):
        for j in range(i, len(mesaj_decriptare), n):
            decriptat += mesaj_decriptare[j]
    return decriptat


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

    try:
        cerinta = int(fisier_intrare.readline())
        mesaj = fisier_intrare.readline().strip()
        m = int(fisier_intrare.readline())

        validare(mesaj, m)                 # apelul functiei de validare
        if cerinta == 1:
            fisier_iesire.write(criptare(mesaj, m))               # apelul functiei de criptare
        elif cerinta == 2:
            fisier_iesire.write(decriptare(mesaj, m))               # apelul functiei de decriptare

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

</syntaxhighlight>