object Main3 extends App {
    var i = 0;
    var j = 0;
    val litery = ('A' to 'Z').toArray;
    var rozmiarTablicy : Int = litery.size;
    var tablica = Array.ofDim[Char](rozmiarTablicy, rozmiarTablicy);

    while (i < rozmiarTablicy) {
        j = 0;
        while (j < rozmiarTablicy) {
            tablica(i)(j) = litery((i + j) % rozmiarTablicy);
            j +=1;
        }
        i += 1;
    }
    /*
    // WYPISYWANIE TABLICY SZYFRUJĄCEJ
    i = 0; 
    j = 0;   
    while (i < rozmiarTablicy) {
        while (j < rozmiarTablicy) {
            print(tablica(i)(j)+" ");
            j += 1;
        }
        print("\n");
        i += 1;
    }
    */
    print("Wpisz haslo do zaszyfrowania (WIELKIMI LITERAMI): ");
    var napis : String = io.StdIn.readLine();
    print("Wpisz klucz (WIELKIMI LITERAMI): ");
    var klucz : String = io.StdIn.readLine();
    var dlugosc : Int = napis.length;
    var dlugoscKlucza : Int = klucz.length;
    var bezSpacji : String = "";

    i = 0;
    while (i < dlugosc) {
        if (napis(i) != ' ') {
            bezSpacji += napis(i);
        }
        i += 1;
    }
    dlugosc = bezSpacji.length;
    
    i = 0;
    while (klucz.length < dlugosc) {
        klucz += klucz(i % dlugoscKlucza);
        i += 1;
    }

    var zaszyfrowane : String = "";
    i = 0;
    while (i < dlugosc) {
        var wiersz : Int = 0;
        while (bezSpacji(i) != tablica(wiersz)(0))
            wiersz += 1;

        var kolumna : Int = 0;
        while (klucz(i) != tablica(0)(kolumna))
            kolumna += 1;
        zaszyfrowane += tablica(wiersz)(kolumna);
        i += 1;
    }
    print("\n\n\n")
    println("Do zaszyfrowania:   " + bezSpacji);
    println("Klucz:              " + klucz);
    println("Zaszyfrowane hasło: " + zaszyfrowane);
    print("\n\n");
}