LeetCode Z 字形变换

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. 先判断给定的行数是否为 1 行或者大于所给字符串的长度,符合上面的条件,最终转换后的结果与原字符串相同,直接返回原字符串;
  2. 题目所提及的变换,其实是遍历字符串,将字符正序(这里表示逐行向下)分配给每行,当到达末行时,方向调转,将字符逆序(这里表示逐行向上)分配给每行,触及首行,方向再次改变,以此往复直到遍历结束;
  3. 先给每行初始化一个空值,遍历行分配值。触及首行或者末行时,方向需要改变(具体见代码)
  4. 最终将所有的元素合并返回。

代码实现


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)

实现结果


实现结果

以上就是本篇的主要内容


欢迎关注微信公众号《书所集录》

你可能感兴趣的:(LeetCode Z 字形变换)