LeetCode Z 字形變換

標(biāo)簽：

Python

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)