// Funkcja do stworzenia tabeli na podstawie podanego klucza
string przetwarzajKlucz(string klucz) {
	
}

// Funkcja do stworzenia tabeli na podstawie podanego klucza
string przetwarzajKlucz(string klucz) {
	/*  
    Stwórzmy używany przez nas alfabet; duże litery
    alfabetu łacińskiego, bez litery Q 
    */
	string alfabet;
	for (char i = 'A'; i <= 'Z'; i++)
		if (i != 'Q')
			alfabet += i;
           
}

// Funkcja do stworzenia tabeli na podstawie podanego klucza
string przetwarzajKlucz(string klucz) {
	/*  
    Stwórzmy używany przez nas alfabet; duże litery
    alfabetu łacińskiego, bez litery Q 
    */
	string alfabet;
	for (char i = 'A'; i <= 'Z'; i++)
		if (i != 'Q')
			alfabet += i;

	for (auto& x : klucz)
		x = toupper(x);
}

// Funkcja do stworzenia tabeli na podstawie podanego klucza
string przetwarzajKlucz(string klucz) {
	/*  
    Stwórzmy używany przez nas alfabet; duże litery
    alfabetu łacińskiego, bez litery Q 
    */
	string alfabet;
	for (char i = 'A'; i <= 'Z'; i++)
		if (i != 'Q')
			alfabet += i;

	for (auto& x : klucz)
		x = toupper(x);

	string przetwarzanyKlucz;
	/*
    Dodajmy do klucza wyjściowego wszystkie znaki,
    bez powtórzeń, jakie mamy w podanym wyrażeniu
    */
	for (auto x : klucz)
		if (alfabet.find(x) != string::npos &&
            	przetwarzanyKlucz.find(x) == string::npos)
			przetwarzanyKlucz += x;

// Funkcja do stworzenia tabeli na podstawie podanego klucza
string przetwarzajKlucz(string klucz) {
	/*  
    Stwórzmy używany przez nas alfabet; duże litery
    alfabetu łacińskiego, bez litery Q 
    */
	string alfabet;
	for (char i = 'A'; i <= 'Z'; i++)
		if (i != 'Q')
			alfabet += i;

	for (auto& x : klucz)
		x = toupper(x);

	string przetwarzanyKlucz;
	/*
    Dodajmy do klucza wyjściowego wszystkie znaki,
    bez powtórzeń, jakie mamy w podanym wyrażeniu
    */
	for (auto x : klucz)
		if (alfabet.find(x) != string::npos &&
            	przetwarzanyKlucz.find(x) == string::npos)
			przetwarzanyKlucz += x;

	// Teraz dopełnijmy klucz pozostałymi literami alfabetu
	for (auto x : alfabet)
		if (przetwarzanyKlucz.find(x) == string::npos)
			przetwarzanyKlucz += x;

	return przetwarzanyKlucz;
}

// Funkcja do zakodowania wiadomości za pomocą klucza
string szyfruj(string wiadomosc, const string klucz) {
    
}

// Funkcja do zakodowania wiadomości za pomocą klucza
string szyfruj(string wiadomosc, const string klucz) { {
	/*
    Upewnijmy się, że tekst do zaszyfrowania
    ma parzystą liczbę znaków
    */
	if (wiadomosc.length() % 2) {
		wiadomosc += 'Z';
	}

	// Interesują nas pary liczb, dlatego i+=2
	for (unsigned int i = 0; i < wiadomosc.length(); i += 2) {
    
    }
}

// Funkcja do zakodowania wiadomości za pomocą klucza
string szyfruj(string wiadomosc, const string klucz) { {
	/*
    Upewnijmy się, że tekst do zaszyfrowania
    ma parzystą liczbę znaków
    */
	if (wiadomosc.length() % 2) {
		wiadomosc += 'Z';
	}

	// Interesują nas pary liczb, dlatego i+=2
	for (unsigned int i = 0; i < wiadomosc.length(); i += 2) {
		/*
        Znajdźmy wiersze i kolumny rozpatrywanych 
        liter w naszym kluczu
        */
		int row[2] = {
			klucz.find(wiadomosc[i]) / 5,
			klucz.find(wiadomosc[i + 1])/ 5
		};
		int col[2] = {
			klucz.find(wiadomosc[i]) % 5,
			klucz.find(wiadomosc[i + 1]) % 5
		};
    }
}

// Funkcja do zakodowania wiadomości za pomocą klucza
string szyfruj(string wiadomosc, const string klucz) { {
	/*
    Upewnijmy się, że tekst do zaszyfrowania
    ma parzystą liczbę znaków
    */
	if (wiadomosc.length() % 2) {
		wiadomosc += 'Z';
	}

	// Interesują nas pary liczb, dlatego i+=2
	for (unsigned int i = 0; i < wiadomosc.length(); i += 2) {
		/*
        Znajdźmy wiersze i kolumny rozpatrywanych 
        liter w naszym kluczu
        */
		int row[2] = {
			klucz.find(wiadomosc[i]) / 5,
			klucz.find(wiadomosc[i + 1])/ 5
		};
		int col[2] = {
			klucz.find(wiadomosc[i]) % 5,
			klucz.find(wiadomosc[i + 1]) % 5
		};
		 
		// Dla dwóch liter w tej samej kolumnie:
        // przesuwamy je o 1 w dół
		if (col[0] == col[1]) {
			wiadomosc[i] = klucz[((row[0]+1)%5)*5 + col[0]];
			wiadomosc[i + 1] = klucz[((row[1]+1)%5)*5 + col[1]];
		}
		// Dla dwóch liter w tym samym rzędzie:
        // przesuwamy je o 1 w prawo
		else if (row[0] == row[1]) {
			wiadomosc[i] = klucz[row[0]*5 + (col[0]+1)%5];
			wiadomosc[i + 1] = klucz[row[1]*5 + (col[1]+1)%5];
		}
		// Dla liter w różnych rzędach i kolumnach:
        // bierzemy litery "na krzyż"
		else {
			wiadomosc[i] = klucz[row[1]*5 + col[0]];
			wiadomosc[i + 1] = klucz[row[0]*5 + col[1]];
		}
	}
	return wiadomosc;
}

// Funkcja do odkodowania zaszyfrowanej wiadomości
string decode(string wiadomosc, const string klucz) {
	return szyfruj(wiadomosc, string(klucz.rbegin(), klucz.rend()));
}

#include <iostream>
#include <string>

using namespace std;

int main() {
	cout << "Podaj klucz do szyfru:" << endl;
	string klucz;
	getline(cin, klucz);
	klucz = przetwarzajKlucz(klucz);

	cout << "Podaj tekst do zaszyfrowania:" << endl;
	string wiadomosc;
	cin >> wiadomosc;
	// Zmieńmy wszystkie litery naszej wiadomości na wielkie
	for (auto& x : wiadomosc)
		x = toupper(x);

	cout << "Do szyfrowania zostanie uzyty nastepujacy klucz:" << endl;
	for (int i = 0; i < 25; i++) {
		cout << klucz[i];
		if ((i+1) % 5 == 0)
			cout << endl;
	}
	cout << endl;

	cout << "Zaszyfrowany tekst to:" << endl;
	cout << (wiadomosc = szyfruj(wiadomosc, klucz)) << endl << endl;

	cout << "Po przetworzeniu przez algorytm odszyfrowania otrzymujemy:" << endl;
	cout << (decode(wiadomosc, klucz)) << endl;

	return 0;
}