leetcode -- 6.Z 字形变换

问题描述：

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时，排列如下：

L   C   I   R
E T O E S I I G
E   D   H   N

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："LCIRETOESIIGEDHN"。

解题要点：找出规律

class Solution(object):
    def convert(self, s, numRows):
        """
        :type s: str
        :type numRows: int
        :rtype: str
        """
        #‘之’字形，4行时从p到l为6个字符，刚好就是a的值（找出的规律）
        a = numRows*2 - 2
        #行数为键，字符为值存入字典dic
        dic = {}
        f = []
        
        #当行数为1，或者大于字符串的长度时，结果为s
        if numRows == 1 or numRows >= len(s):
            return s
        
        #遍历字符串
        for i in range(len(s)):
            #b为i整除a的余数
            b = i % a
            #b小于行数时按正常的顺序
            if b < numRows:
                dic.setdefault(b, [ ]).append(s[i])
            #上边这行代码的意思是若字典中有这一个key的话就直接append字符
            #没有的话则新建该key，并vale为一个[ ]
            #大于行数时按规律求得其所属的行数
            else:
                d = a - b
                dic.setdefault(d, [ ]).append(s[i])
        #将字典按key把其对应的vale加在一起整合成一个列表
        for i in range(numRows):
            f += dic[i]
        #将列表里的元素转换成字符串
        g = ''.join(f)
        return g