6. N 字形变换 @Python @Go

原创已于 2025-09-12 15:54:55 修改 · 219 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #go

于 2023-08-07 19:42:38 首次发布

Leetcode 专栏收录该内容

126 篇文章

订阅专栏

博客围绕LeetCode上的Zigzag Conversion题目展开，介绍解题思路为字符串处理，用方向变量标记方向，分析复杂度为时间O(n)、空间O(n)，并给出Python和Java代码实现。

原题

https://leetcode.cn/problems/zigzag-conversion/

思路

字符串处理，用方向变量来标记方向。

复杂度

时间：O(n)
空间：O(n)

Python代码

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        l = ["" for i in range(numRows)]
        i = 0
        direction = 1
        for ch in s:
            l[i] += ch
            if i + direction >= numRows or i + direction < 0:
                direction *= -1
            i += direction
        return "".join(l)

Go代码

func convert(s string, numRows int) string {
    if numRows == 1 {
        return s
    }
    l := make([]string, numRows)
    i := 0 
    direction := 1 
    for _, ch := range s {
        l[i] += string(ch)
        if i + direction >= numRows || i + direction < 0 {
            // 改变方向
            direction *= -1
        }
        i += direction
    }
    // 拼接字符串
    return strings.Join(l, "") 
}