leetcode--z字形转换

1.题目

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

比如输入字符串为 "PAYPALISHIRING" 行数为 3 时

例如：

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

思路：

解题思路：
特判，若行数numRows<2numRows<2，返回ss
初试化res=["",\cdots,""]res=["",⋯,""]，长度为numRowsnumRows，res[i]res[i]表示第ii行存放的字符。
初始化标志flag=-1flag=−1，初试行i=0i=0
遍历字符串中每一个字符cc：
若i==0\ or\ i==numRows-1i==0ori==numRows−1，说明应该更换方向，向下为11，向上为-1−1，因此flag=-flagflag=−flag
将当前字符cc填入对应行ii。
更新行i=i+flagi=i+flag
将resres中各行连接起来并返回。

代码如下：

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if(numRows<2):
            return s
        res=[""]*numRows
        n=len(s)
        flag=-1
        i=0
        for c in s:
            res[i]+=c
            if(i==0 or i==numRows-1):flag=-flag
            i+=flag
        return "".join(res)
#自己测一下
if __name__ == '__main__':
    s = Solution()
    result = s.convert(s = "PAYPALISHIRING", numRows = 3)
    print('z字形转换',result)

结果如下：

z字形转换 PAHNAPLSIIGYIR