Leetcode刷题6— Z 字形变换
问题描述
将一个给定字符串 s
根据给定的行数 numRows
,以从上往下、从左到右进行 Z 字形排列。
排列完成后,从左往右逐行读取字符,生成新的字符串。
示例
示例 1:
“”"
输入: s = “PAYPALISHIRING”, numRows = 3
输出: “PAHNAPLSIIGYIR”
解释:
P A H N
A P L S I I G
Y I R
“”"
示例 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 . l e n g t h ≤ 1000 1 \leq s.length \leq 1000 1≤s.length≤1000
s
由英文字母(小写和大写)、','
和'.'
组成。- 1 ≤ n u m R o w s ≤ 1000 1 \leq numRows \leq 1000 1≤numRows≤1000
方法:按行访问
思路:
- 如果
numRows == 1
,直接返回字符串s
。 - 使用一个列表存储每一行的字符。
- 遍历字符串时,按 Z 字形的顺序将字符加入到对应的行:
- 用变量
current_row
表示当前行,初始值为 0。 - 用变量
going_down
表示当前的方向(向下或向上)。 - 如果当前行是第一行或最后一行,改变方向。
- 用变量
- 遍历完成后,将所有行的字符合并,形成最终结果。
代码实现:
def convert(s: str, numRows: int) -> str:if numRows == 1:return srows = [""] * min(numRows, len(s))current_row = 0going_down = Falsefor char in s:rows[current_row] += charif current_row == 0 or current_row == numRows - 1:going_down = not going_downcurrent_row += 1 if going_down else -1return "".join(rows)