#include <iostream>

using namespace std;

int szyfrowanie(string wyraz){

char znak;
    int ile=wyraz.length();

    for(int i=0; i<ile; i++){
        znak=wyraz[i];
        znak=toupper(znak);

        if (znak==32) znak=' ';
        else if (znak=='X') znak='A';
        else if (znak=='Y') znak='B';
        else if (znak=='Z') znak='C';
        else znak+=3;
        cout<<znak;
    }
}

int odszyfrowanie(string wyraz){

char znak;
    int ile=wyraz.length();

    for(int i=0; i<ile; i++){
        znak=wyraz[i];
        znak=toupper(znak);

        if (znak==32) znak=' ';
        else if (znak=='A') znak='X';
        else if (znak=='B') znak='Y';
        else if (znak=='C') znak='Z';
        else znak-=3;

        cout<<znak;
    }
}

int main()
{
    string wyraz;
    int wybor;

    cout<<"Podaj wyraz do szyfrowania: "<<endl;
    getline(cin,wyraz);

    cout<<" __________________ "<<endl;
    cout<<"|                  |"<<endl;
    cout<<"| 1. Szyfrowanie   |"<<endl;
    cout<<"| 2. Odszyfrowanie |"<<endl;
    cout<<"|__________________|"<<endl;

    cin>>wybor;

    switch (wybor){
    case 1:
        szyfrowanie(wyraz);
        break;
    case 2:
        odszyfrowanie(wyraz);
        break;
    default:
        cout<<"Nie ma takiego wyboru";
        break;
    }
    return 0;
}