LeetCode刷题笔记第6题:Z字形变换
想法:
要完成字符串根据给定的行数从上往下,从左到右完成Z字形排列。当只有一行时直接返回原字符串,当行数大于1时,先以行数构建一个行数数值个空字符串的列表,表示将排列之后的每行字符存入对应位置的列表的位置中。因为在排列的时候,元素的列数在遇到Z字拐角处递增和递减交替进行,用flag来控制列数的递增和递减。详细注释见代码。
class Solution:
def convert(self, s: str, numRows: int) -> str:
# 当给定行数为1时,返回原字符串
if numRows < 2:
return s
# 以列数构建列表中的位置数,为之后存放每行数据做准备
res = ["" for _ in range(numRows)]
# 设置初始值,flag用于Z字形排列在拐角处的操作
i, flag = 0, -1
for c in s:
# 将每行的数据加入列表对应的位置处
res[i] += c
# 此处判断完成Z字形排列的拐角转行
if i == 0 or i == numRows - 1: flag = -flag
i += flag
# 将每行的字符串(每个位置字符串)拼接
return "".join(res)