题目
将一个给定字符串 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
由英文字母(小写和大写)、','
和'.'
组成 1 <= numRows <= 1000
解题思路
避免直接字符串拼接,使用一个字符串列表来代替。通过切换方向变量going_down
和当前行索引cur_row
来模拟Z字形排列。
代码
class Solution:
def convert(self, s: str, numRows: int) -> str:
if numRows == 1 or numRows >= len(s):
return s
rows = [''] * min(numRows, len(s))
cur_row = 0
going_down = False
for c in s:
rows[cur_row] += c
if cur_row == 0 or cur_row == numRows - 1:
going_down = not going_down
cur_row += 1 if going_down else -1
return ''.join(rows)
运行
sol = Solution()
print(sol.convert("PAYPALISHIRING", 3)) # 应输出 "PAHNAPLSIIGYIR" print(sol.convert("PAYPALISHIRING", 4)) # 应输出 "PINALSIGYAHRPI"
总结
Z 字形变换算法通过模拟字符在矩阵中的垂直移动来实现。算法首先创建一个字符串列表,每个字符串代表矩阵的一行。遍历输入字符串中的每个字符,根据当前的行索引将其添加到对应的行字符串中。当字符移动到第一行或最后一行时,改变其垂直移动的方向。这种方法可以有效地按照 Z 字形模式来组织字符,然后通过顺序连接每一行得到最终的结果字符串。这种实现避免了对字符串进行多次不必要的重复拼接,从而提高了算法的执行效率。