【LeetCode】6.Z字形变换(python版)

最新推荐文章于 2022-04-20 16:03:25 发布

一颗随风而倒的墙头草

最新推荐文章于 2022-04-20 16:03:25 发布

阅读量440

点赞数 1

分类专栏： LeetCode 思想奇特

本文链接：https://blog.youkuaiyun.com/qq_20141867/article/details/104002251

版权

该博客介绍了如何使用Python解决LeetCode的6题——Z字形变换。文章通过两种方法详细解析了算法思路，包括利用二维数组模拟Z字形排列过程和通过计算每个位置的字符索引来直接获取排列后的字符串。并提供了示例来帮助理解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：
将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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

思路：

方法一：

我们想象一下由字符串构造z形图案的过程，如图所示，假设我们用四个数组分别收集每一行的字符，用当前行(绿箭头)和当前方向(红箭头)指示。只有当红箭头遇到边界，也就是当字符排到第一行或者最后一行时，箭头才会反向。当红箭头向下时，绿箭头指示的所在行数依次增大；当红箭头向上时，绿箭头指示的所在行数依次减小。

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows == 1:
            return s
        # 这里采用字符串保存每一行的字符，而不是数组，可以减少一步由数组到字符串的转化
        rows = [''] * min(numRows, len(s))
        # 图中的绿箭头，指示所在行数
        cur_row = 0
        # 图中的红箭头，指示方向
        going_down = False
        for c in s:
            rows[cur_row] += c
            # 当箭头到达边界时，转向
            if cur_row == 0 or cur_row == numRows - 1:
                going_down = not going_down
            # 如果箭头方向为下，行数指针增大，如果箭头方向为上，行数指针减小
            cur_row += 1 if going_down else -1
        res = ''.join(rows)
        return res