LeetCode——Z字形变换 by python

题目：

 

将字符串 "PAYPALISHIRING" 以Z字形排列成给定的行数：

P   A   H   N
A P L S I I G
Y   I   R

之后从左往右，逐行读取字符："PAHNAPLSIIGYIR"

实现一个将字符串进行指定行数变换的函数:

string convert(string s, int numRows);

示例 1:

输入: s = "PAYPALISHIRING", numRows = 3
输出: "PAHNAPLSIIGYIR"

示例 2:

输入: s = "PAYPALISHIRING", numRows = 4
输出: "PINALSIGYAHRPI"
解释:

P     I    N
A   L S  I G
Y A   H R
P     I

分析：

刚开始看题目的时候被两列之间的空格给迷惑了，后面看了网上其他人的分析，知道了不需要管空格。这个采用二维数组，然后慢慢填充就行了，一次遍历填充一行中的某一个位置，然后换到下一行。

代码：

ef convert(s, numRows):
    """
    :type s: str
    :type numRows: int
    :rtype: str
    """
    if(numRows == 1): return s
    a = [[]for i in range(numRows)]#产生一个有numrows单元的list，每个list可以在后面添加元素
    r = 0
    direct = 1 #行前进的方向是向上还是向下
    for c in s:
        a[r].append(c)
        if r >=numRows-1:
            direct = -1
        elif r == 0:
            direct = 1
        r +=direct
    answer = ""
    for row in a:#按行优先打印出来
        for col in row:
            answer += col
    return answer
if __name__ == "__main__":
    print(convert("PAYPALISHIRING",4))