Editing 3142 - Polybius


= Polybius =
Cifrul Polybius este un cifru folosit de Grecii Antici care are la bază un tabel de <code>5</code> linii și <code>5</code> coloane pentru a cripta sau decripta un șir de caractere. Pentru criptare, luăm fiecare caracter din șir și afișăm linia și coloana din tabel unde se află caracterul. Pentru decriptare afișăm caracterul care se află în tabel la linia și coloana dată

Împreună cu șirul <code>INF</code>, afișăm <code>243321</code> deoarece litera <code>I</code> se află pe poziția <code>(2, 4)</code>, litera <code>N</code> se află pe poziția <code>(3, 3)</code> iar litera <code>F</code> se află pe poziția <code>(2, 1)</code>.

= Cerința =
Se dă un număr <code>p</code>, un șir de caractere <code>t</code> de <code>25</code> de caractere corespunzător tabelului și un șir de caractere <code>s</code>. Să se determine:

1. Pentru <code>p = 1</code>, să se cripteze șirul <code>s</code>, format doar din litere mari, folosind tabelul <code>t</code>.

2. Pentru <code>p = 2</code>, să se decripteze șirul <code>s</code>, format doar din cifre, folosind tabelul <code>t</code>.

= Date de intrare =
Fișierul de intrare <code>polybius.in</code> conține pe prima linie numărul <code>p</code>, pe a doua linie șirul <code>t</code> format din <code>25</code> de caractere, reprezentând toate elementele tabelului, de la stânga la dreapta și de sus în jos. A treia linie șirul de litere <code>s</code> care trebuie criptat, dacă <code>p=1</code>, respectiv șirul de cifre care trebuie decriptat, dacă <code>p=2</code>.

= Date de ieșire =
Pentru <code>p = 1</code>, fișierul de ieșire <code>polybius.out</code> va conține pe prima linie rezultatul criptării. Cifrele se afișează fără spațiu între ele.

Pentru <code>p = 2</code>, fișierul de ieșire <code>polybius.out</code> va conține pe prima linie rezultatul decriptării.

= Restricții și precizări =

* <code>1 ≤ p ≤ 2</code>
* lungimea șirului <code>t</code> este de <code>25</code> de caractere
* <code>1 ≤</code> lungimea șirului de litere <code>s ≤ 500.000</code>
* Se garantează că toate caracterele din <code>s</code> sunt incluse în șirul <code>t</code> și caracterele din <code>t</code> sunt distincte două câte două.
* O literă din alfabet va lipsi întotdeauna din tabel, fiind un tabel <code>5x5</code> iar alfabetul are <code>26</code> de caractere.

= Exemplul 1: =
<code>polybius.in</code>
 1
 ABCDEFGHIKLMNOPQRSTUVWXYZ 
 ANAAREMERE
<code>polybius.out</code>
 11331111421532154215

= Exemplul 2: =
<code>polybius.in</code>
 2
 ABCDEFGHIKLMNOPQRSTUVWXYZ 
 11331111421532154215
<code>polybius.out</code>
 ANAAREMERE

== Încărcare soluție ==

=== Lipește codul aici ===
<syntaxhighlight lang="python" line="1">
import sys

cer = int(input())
a = input().strip()
k = 0
mat = [['' for _ in range(6)] for _ in range(6)]
for i in range(1, 6):
    for j in range(1, 6):
        while a[k] == ' ':
            k += 1
        mat[i][j] = a[k]
        k += 1

sys.stdin.readline()
X = [0] * 30
if cer == 2:
    while True:
        line = sys.stdin.readline().strip()
        if not line:
            break
        a, b = line.split()
        print(mat[int(a)][int(b)], end='')
else:
    while True:
        line = sys.stdin.readline().strip()
        if not line:
            break
        ch = line[0]
        print(X[ord(ch) - ord('A')], end='')


</syntaxhighlight>