Z 字形变换
题目来源：https://leetcode-cn.com/problems/zigzag-conversion/
题目
将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时，排列如下：
L C I R
E T O E S I I G
E D H N之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："LCIRETOESIIGEDHN"。
请你实现这个将字符串进行指定行数变换的函数：
string convert(string s, int numRows);
示例 1:
输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"示例 2:
输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"
解释:
L D R
E O E I I
E C I H N
T S G解题思路
- 先判断给定的行数是否为 1 行或者大于所给字符串的长度，符合上面的条件，最终转换后的结果与原字符串相同，直接返回原字符串；
- 题目所提及的变换，其实是遍历字符串，将字符正序（这里表示逐行向下）分配给每行，当到达末行时，方向调转，将字符逆序（这里表示逐行向上）分配给每行，触及首行，方向再次改变，以此往复直到遍历结束；
- 先给每行初始化一个空值，遍历行分配值。触及首行或者末行时，方向需要改变（具体见代码）
- 最终将所有的元素合并返回。
代码实现
class Solution:
def convert(self, s: str, numRows: int) -> str:
'''将字符串进行 Z 字形排列
Args:
s: 给定的字符串
numRows: 按多少行进行 Z 字形排列
Returns:
返回 Z 字形排列的结果
'''
# 给定列数为 1，或者大于等于字符串长度直接返回
if numRows == 1 or numRows >= len(s):
return s
# 排列时，只有正序到最后一列，或者逆序到第一列时，方向才会改变
i, sign = 0, False
# 先给每行一个初始值
lst = ['' for _ in range(numRows)]
# 遍历字符串，逐行添值
for ch in s:
lst[i] += ch
# 判断方向是否需要改变
# 以向下为正向，向上会逆向
# 触及最顶行，最末行都需改变位置
if i == 0 or i == numRows - 1:
sign = not sign
i += 1 if sign else - 1
# 合并列表返回合并后的字符串
return ''.join(lst)实现结果
以上就是本篇的主要内容
欢迎关注微信公众号《书所集录》