2005-ös érettségi - Vigenére

vigenere.rb

H=("ÖÜÓŐÚŰÁÉÍ".split(//).zip("OUOOUUAEI".split(//))).to_h

t=('A'..'Z').to_a

tt={}
tt.merge!({t[0]=>t.zip(t).to_h})
for p in 0..t.size-2 do 
  tt.merge!({t[p+1]=>t.zip(t[p+1..-1]+t[0..p]).to_h})
end

kodolatlan="EZAPROBASZOVEGAMITKODOLUNK" #STDIN.readline.chomp
kodolatlan=kodolatlan.gsub(" ","").upcase.split(//).map{|c| H[c]==nil ? c : H[c] }.join 
kodszo="auto" #STDIN.readline.chomp
kodszo=kodszo.gsub(" ","").upcase.split(//).map{|c| H[c]==nil ? c : H[c] }.join
puts kodszo
kodolt=kodolatlan.split(//).collect.with_index.map{|c,i| 

    tt[kodszo[i % kodszo.size]][c]
}.join
puts kodolt