深入理解 LeetCode 第6题：ZigZag Conversion，轻松斩获「中等」难度成就

2024-02-18 11:44:20

导语

LeetCode 是全球知名的算法题竞赛平台，汇集了大量难度各异的算法题。每道题都对应一个具体的编程问题，用户需要根据题意，使用编程语言编写代码来解决问题。LeetCode 题目的难度分为简单、中等和困难三个等级，其中中等难度的题目往往更具挑战性，需要更深入的思考和编程技巧。

题意解析

LeetCode 第 6 题：ZigZag Conversion 的题意如下：

给定一个字符串 s 和一个整数numRows，将字符串 s 按numRows行书写，并将每一行的字符串连接起来。

例如，s = "PAYPALISHIRING" 和numRows = 3，输出为 "PAHNAPLSIIGYIR"。

算法思路

这道题虽然是中等难度，但实际上解起来很简单。根据题目要求，我们需要把原字符串按一定规则拆成 n 个子字符串，再将这 n 个子字符串首尾相连即可。

观察示例可以发现，字符串中的每个字符都属于某个子字符串，而每个子字符串中的字符按照一定的顺序排列。因此，我们可以为原字符串中的每个字符标记其所属的子字符串的序号，然后再根据该序号来构建 n 个子字符串，最后将它们连接起来即可。

具体来说，我们可以使用一个数组numRows来存储每个子字符串的起始字符索引，然后遍历字符串s，将每个字符添加到相应的子字符串中。最后，将 n 个子字符串连接起来，即可得到最终结果。

代码实现

def convert(s, numRows):
    """
    :type s: str
    :type numRows: int
    :rtype: str
    """
    if numRows == 1 or numRows >= len(s):
        return s

    numRows = min(numRows, len(s))
    rows = [[] for _ in range(numRows)]
    row, direction = 0, 1

    for char in s:
        rows[row].append(char)
        if row == 0:
            direction = 1
        elif row == numRows - 1:
            direction = -1
        row += direction

    result = ""
    for row in rows:
        result += "".join(row)

    return result