LeetCode6-Z字形变换

题目

将一个给定字符串根据给定的行数numRows，以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时，排列如下：

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

思路

利用长度为numRows的数组，第i个元素表示第i行包含的字符。利用flag控制Z字前进方向，如果当前在第一行或最后一行则改变Z字前进方向。

代码

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows<2:return s
        result=["" for i in range(numRows)]#创建长度为numRows的空数组
        flag,i=-1,0
        for a in s:#遍历字符串
            result[i]+=a#第i行加上对应字符
            if i == 0 or i == numRows-1:#改变Z字前进方向
                flag=-flag
            i+=flag
        return "".join(result)#数组合成字符串