LeetCode0054.螺旋矩阵 Go语言AC笔记

时间复杂度:O(n²)

解题思路

我更喜欢把这种解题方法称为“剥洋葱”法。从外向内按照一圈一圈的方法顺时针遍历二维数组中的所有元素,外面一圈剥完后就扔掉,让次外圈暴露出来成为新的外圈,直到最内圈的洋葱心暴露出来,遍历完所有的元素。

具体实现涉及到四个限制变量——top、bottom、left和right,分别代表着最外圈的上边界、下边界、左边届和右边届,还需要一个变量cnt记录还没有遍历的元素数量。这样只要cnt不为0,就说明还没有遍历完所有元素,那就按照先从左向右、再从上到下、再从右到左、最后从下到上顺时针的顺序遍历最外圈的每一层元素,每遍历完一层,就更新一次边界。

LeetCode0054.螺旋矩阵 Go语言AC笔记_第1张图片

上图可以很好地帮助理解top、bottom、left和right的含义。

AC代码

func spiralOrder(matrix [][]int) []int {
    res:=[]int{}
    m,n:=len(matrix),len(matrix[0])
    if m==0||n==0{
        return res
    }
    top,bottom,left,right:=0,m-1,0,n-1
    cnt:=m*n
    for cnt>0{
        for i:=left;i<=right&&cnt>0;i++{
            res=append(res,matrix[top][i])
            cnt--
        }
        top++
        for i:=top;i<=bottom&&cnt>0;i++{
            res=append(res,matrix[i][right])
            cnt--
        }
        right--
        for i:=right;i>=left&&cnt>0;i--{
            res=append(res,matrix[bottom][i])
            cnt--
        }
        bottom--
        for i:=bottom;i>=top&&cnt>0;i--{
            res=append(res,matrix[i][left])
            cnt--
        }
        left++
    }
    return res
}

感悟

题解的代码看起来真别扭,还是这个AC代码开起来好理解。

你可能感兴趣的:(算法,数据结构,golang,力扣,leetcode)