54. 螺旋矩阵 (leetcode)

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

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

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

来源:https://leetcode-cn.com/problems/spiral-matrix

打卡:

class Solution:
    def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
        l=[]
        i,j,di,dj=0,0,0,1
        for k in range(len(matrix)*len(matrix[0])):
            print(matrix[i][j])
            l.append(matrix[i][j])
            matrix[i][j]=None
            if i+di>=len(matrix) or j+dj>=len(matrix[0]):
                di,dj=dj,-di       
            elif matrix[i+di][j+dj]==None:
                di,dj=dj,-di 
            j+=dj
            i+=di
        return l

猜你喜欢

转载自blog.csdn.net/Yao_June/article/details/112795984