7. kotprog, Python verzio 1

#!/usr/bin/env python

def encode(text, key):
    table = []
    cols = len(key.split(' '))
    for i in range(0, len(text) - 1, cols):
        table.append(text[i:i + cols])

    result = ""
    for i in range(0, cols):
        for j in range(0, len(table)):
            result += table[j][int(key.split(' ')[i]) - 1]
    return result

def decode(text, key, cols):
    pass

infile = open("be.txt")

func, cols = infile.readline().split(' ')
key = infile.readline()
text = infile.readline()

if func == '1':
    text = encode(text, key)
else:
    text = decode(text, key)

outfile = open("ki.txt", "w")
outfile.write(text + "\n")
outfile.close()