leedcode 螺旋矩阵II

最新推荐文章于 2024-09-24 16:24:36 发布

原创最新推荐文章于 2024-09-24 16:24:36 发布 · 74 阅读

0 ·

CC 4.0 BY-SA版权

leedcode 专栏收录该内容

50 篇文章

订阅专栏

本文介绍了一种生成螺旋矩阵的方法，通过将读取数据操作改为写入数据操作，实现了从外向内螺旋填充数字的过程。该算法使用了四个边界变量来控制填充范围，并在每轮填充后更新边界。

和螺旋矩阵类似，是需要把读取数据的操作改成写入数据的操作即可

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        result = []
        for i in range(n):
            temp = []
            for j in range(n):
                temp.append(0)
            result.append(temp)
        x1, y1, x2, y2 = 0, 0, n - 1, n - 1
        count = 1
        while x1 <= x2 and y1 <= y2:
            for i in range(x1, x2 + 1):
                result[y1][i] = count
                count += 1
            for j in range(y1 + 1, y2 + 1):
                result[j][x2] = count
                count += 1
            if x1 < x2 and y1 < y2:
                for i in range(x2 - 1, x1, - 1):
                    result[y2][i] = count
                    count += 1
                for j in range(y2, y1, -1):
                    result[j][x1] = count
                    count += 1
            x1 += 1
            y1 += 1
            x2 = x2 - 1
            y2 = y2 - 1
        return result