Encryption

题目

An English text needs to be encrypted using the following encryption scheme.
附上链接： Encryption

一点小知识（智障的小知识）

但想要判断列表等可迭代对象的索引是否越界，不可以直接使用
if s[i] is not None
因为如果索引已经越界的话，这样的一句话会报错，而不是正常实现想要的效果
我个人的解决办法是采用下面这句话
if i < len(s)
其实没什么好写的，只是我经常犯这样的错误。。。所以记录一下

其实没什么好写的

这个题并不难，但我却用了近半个小时才完成，我之所以会写它就是为了提醒自己做事一定要认真，一定要把情况考虑全面，不求一次通过，也要减少修改次数

最后给出代码

def encryption(s):
    row = int(math.floor(math.sqrt(len(s))))
    col = int(math.ceil(math.sqrt(len(s))))
    if row * col < len(s):
        row += 1
    k = 0
    a = []
    for i in range(row):
        b = []
        for j in range(col):
            if k < len(s):
                b.append(s[k])
                k += 1
        a.append(b)
    c = ''
    for i in range(col):
        for j in range(row):
            if i < len(a[j]):
                c += a[j][i]
        c += ' '
    return c

这个代码很冗杂，如果有大佬能提出一些改进意见将不胜感激