#include <iostream>
using namespace std;

int max_cifra(int broj)
{
    int ret = 0, cifra;

    while (broj != 0)
    {
       cifra = broj % 10;

       if(cifra > ret)
       {
            ret = cifra;
       }

       broj = broj /10;
    }

    return (ret);
}

string kodiraj (string s, int kod)
{
    int max_br = max_cifra(kod);
    string ret = s;


    for(int i=0; i < s.length(); i++)
    {
        if(s[i] == ' ')
            continue;

        ret[i] = s[i] + max_br;

        if(ret[i] > 'z')
            ret[i] -= 26;
    }

    return (ret);
}

string dekodiraj (string s, int kod)
{
    int max_br = max_cifra(kod);
    string ret = s;


    for(int i=0; i < s.length(); i++)
    {
        if(s[i] == ' ')
            continue;

        ret[i] = s[i] - max_br;

        if(ret[i] < 'a')
            ret[i] += 26;
    }

    return (ret);
}

int main()
{
    string s;
    int kod, izbor;
    char t;

    cout << "1. Kodiraj" << endl;
    cout << "2. Dekodiraj" << endl;
    cout << "> ";
    cin >> izbor;

    switch(izbor)
    {
        case 1:
            cout << "Vnesi rechenica: ";
            getline(cin, s); // reads newline koga kje vnesam broj i chuknam ENTER
            getline(cin, s); // reads actual string sho kje se koristi
            cout << "Vnesi kod: ";
            cin >> kod;
            cout << s << " -> " << kodiraj(s, kod) << endl;
            break;

        case 2:
            cout << "Vnesi rechenica: ";
            getline(cin, s); // reads newline koga kje vnesam broj i chuknam ENTER
            getline(cin, s); // reads actual string sho kje se koristi
            cout << "Vnesi kod: ";
            cin >> kod;
            cout << s << " -> " << dekodiraj(s, kod) << endl;
            break;

        default:
            cout << "Vnesovte nevaliden izbor!" << endl;
    }


    return(0);
}