Leetcode 54.螺旋矩阵（Spiral Matrix）

最新推荐文章于 2024-09-25 13:54:13 发布

就叫昵称吧

最新推荐文章于 2024-09-25 13:54:13 发布

阅读量186

点赞数

分类专栏： Leetcode 文章标签：算法 leetcode

本文链接：https://blog.csdn.net/qq_39378221/article/details/104460402

版权

Leetcode 专栏收录该内容

595 篇文章 0 订阅

订阅专栏

Leetcode 54.螺旋矩阵

1 题目描述（Leetcode题目链接）

给定一个包含 m x n 个元素的矩阵（m 行, n 列），请按照顺时针螺旋顺序，返回矩阵中的所有元素。

输入:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
输出: [1,2,3,6,9,8,7,4,5]

输入:
[
  [1, 2, 3, 4],
  [5, 6, 7, 8],
  [9,10,11,12]
]
输出: [1,2,3,4,8,12,11,10,9,5,6,7]

2 题解

第一想法就是上下左右走，把走过的位置置零或置其它数，绕圈走，越界或走到之前走的位置就拐弯，所以我用了四个变量来记录上下左右几个方向，写出来的题解冗得一批。

class Solution:
    def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
        if not matrix:
            return []
        m = len(matrix)
        n = len(matrix[0])
        retv = []
        i, j = 0, 0
        right, down, left, up = 1, 0, 0, 0
        for _ in range(m*n):
            retv.append(matrix[i][j])
            matrix[i][j] = 0
            if right:
                if j + 1 == n or matrix[i][j + 1] == 0:
                    right, down = 0, 1
                    i += 1
                else:
                    j += 1
            elif down:
                if i + 1 == m or matrix[i + 1][j] == 0:
                    down, left = 0, 1
                    j -= 1
                else:
                    i += 1
            elif left:
                if j - 1 < 0 or matrix[i][j - 1] == 0:
                    left, up = 0, 1
                    i -= 1
                else:
                    j -= 1
            elif up:
                if i - 1 < 0 or matrix[i - 1][j] == 0:
                    up, right = 0, 1
                    j += 1
                else:
                    i -= 1
        return retv

评论区里面的解答同样是改方向，但是非常简洁，贴过来参考，他用di和dj两个变量控制方向，很巧妙。具体含义可以观察：
向右： $d i = 0, d j = 1$
向下： $d i = 1, d j = 0$
向左： $d i = 0, d j = - 1$
向上： $d i = - 1, d j = 0$
可以观察到，每次转方向的时候都是 $d i, d j = d j, - d i$

class Solution:
    def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
        r, i, j, di, dj = [], 0, 0, 0, 1
        if matrix != []:
            for _ in range(len(matrix) * len(matrix[0])):
                r.append(matrix[i][j])
                matrix[i][j] = 0
                if matrix[(i + di) % len(matrix)][(j + dj) % len(matrix[0])] == 0:
                    di, dj = dj, -di
                i += di
                j += dj
        return r