LeetCode 6.Z字形变换

最新推荐文章于 2025-05-20 08:09:36 发布

Heartache.246

最新推荐文章于 2025-05-20 08:09:36 发布

阅读量261

点赞数 9

分类专栏：力扣刷题文章标签： leetcode 算法 python

本文链接：https://blog.youkuaiyun.com/qq_74747823/article/details/142565361

版权

力扣刷题专栏收录该内容

25 篇文章

订阅专栏

LeetCode 6.Z字形变换

将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：

P A H N
A P L S I I G
Y I R

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

示例 1：

输入：s = “PAYPALISHIRING”, numRows = 3
输出：“PAHNAPLSIIGYIR”

示例 2：

输入：s = “PAYPALISHIRING”, numRows = 4
输出：“PINALSIGYAHRPI”
解释：
P I N
A L S I G
Y A H R
P I

示例 3：

输入：s = “A”, numRows = 1
输出：“A”

提示：

1 <= s.length <= 1000
s 由英文字母（小写和大写）、',' 和 '.' 组成
1 <= numRows <= 1000

代码：

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        # 如果numRows为1，说明不需要重新排列，直接返回原始字符串
        if numRows == 1: 
            return s
        
        # 初始化一个列表，用来存储每一行的字符
        rows = [""] * numRows
        
        # 计算Z字形的周期，即一个完整的Z字形需要的行数
        n = 2 * numRows - 2
        
        # 遍历字符串s中的每一个字符
        for i, char in enumerate(s):
            # 计算当前字符应该放置的行
            # x = i % n 是当前字符索引对周期的取余，得到的值表示当前字符在Z字形中的相对位置
            # min(x, n - x) 确保当x大于numRows-1时，字符放置在下行，因为Z字形是对称的
            x = i % n
            rows[min(x, n - x)] += char
        
        # 将所有行的字符拼接起来，形成最终的字符串
        return "".join(rows)