LeetCode 力扣题解（6）Z 字形变换

LeetCode 力扣题解（6）Z 字形变换，实现语言 Python，解决方法：列表遍历。

一、题目：Z 字形变换

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

比如输入字符串为 “PAYPALISHIRING” 行数为 3 时，排列如下：

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

示例 3：

输入：s = "A", numRows = 1
输出："A"

提示：

• $1<=s.length<=1000$
• $s由英文字母（小写和大写）{、}^{\prime }{,}^{\prime }{和}^{\prime }{.}^{\prime }组成$
• $1<=numRows<=1000$

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zigzag-conversion
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、题解：列表遍历

列表遍历，使用列表遍历方法，只需要对字符串 s 遍历一次，所以时间复杂度是 $O(n)$，空间复杂度是 $O(n)$。

从示例可以看出，按照字符串的顺序，字符会一个一个从 0 行往下排，排到 numRows - 1 行之后再一个一个往上排到 0 行。

PAHNAPLSIIGYIR

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

s_list[i]
s_list[0]	P	A	H	N
s_list[1]	A	P	L	S	I	I	G
s_list[2]	Y	I	R

所以，先创建 numRows 个列表 s_list[0:numRows]，接着一个一个字符添加进去：

• i = 0，x = -1
• s_list[i] = s_list[i] + c
• 如果 i==0 或者 i==numRows-1，x = -x
• i = i + x

如此循环，直到遍历完所有字符，然后按顺序合并 s_list 字符串并返回即可。

三、示例代码

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        s_list = ['' for _ in range(numRows)]
        i = 0
        x = -1

        if numRows < 2:
            return s
            
        for c in s:
            s_list[i] += c
            if i==0 or i==numRows-1:
                x = -x
            i += x

        return ''.join(s_list)