LeetCode算法题6：Z 字形变换解析

最新推荐文章于 2025-05-23 17:46:09 发布

哪得小师弟

最新推荐文章于 2025-05-23 17:46:09 发布

阅读量993

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Leetcode算法题分析

本文链接：https://blog.youkuaiyun.com/x603560617/article/details/84303612

Leetcode算法题分析专栏收录该内容

156 篇文章

订阅专栏

本文介绍了一种将字符串按照Z字形排列并重新组合的算法。通过示例详细解析了如何根据给定的行数，将字符串LEETCODEISHIRING转换成特定的Z字形排列，并提供了C++及Python3的实现代码。

摘要生成于 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

这个题就是一个找规律的题，每一行的字母索引都是按步可找的。所以找一下规律先，其实看示例2，第一行就是每个元素之间都差2*numRows-2，最后一行也是这样，中间的两行一部分元素是这样，还有一部分元素与行数有关。所以程序就可以直接编写了：
C++源代码：

class Solution {
public:
    string convert(string s, int numRows) {
        if (numRows == 1) return s;
        string str;
        int n = s.length();
        int step = 2 * numRows - 2;

        for (int i = 0; i < numRows; i++) 
        {
            for (int j = 0; j + i < n; j += step) 
            {
                str += s[j + i];
                if (i != 0 && i != numRows - 1 && j + step - i < n)
                    str += s[j + step - i];
            }
        }
        return str;
    }
};

python3源代码：

class Solution:
    def convert(self, s, numRows):
        """
        :type s: str
        :type numRows: int
        :rtype: str
        """
        if numRows==1:
            return s
        ns = ""
        n = len(s)
        step = 2 * numRows - 2
        for i in range(numRows):
            for j in range(i, n, step):
                ns += s[j]
                if i!=0 and i!=numRows-1 and j+step-2*i<n:
                    ns += s[j+step-2*i]
        return ns