Leetcode(Java)-54. 螺旋矩阵

给定一个包含 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]

class Solution {
    public List<Integer> spiralOrder(int[][] matrix) {
        int m = matrix.length;
        if(m==0)
            return new ArrayList<Integer>();
        int n = matrix[0].length;
        boolean[][] visited = new boolean[m+2][n+2];
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=n;j++)
            {
                visited[i][j] = true;
            }
        }
        int[] dx = {0,1,0,-1};
        int[] dy = {1,0,-1,0};
        List<Integer> res = new ArrayList<>();
        int x=1;
        int y=1;
        while(res.size()<m*n)
        {
            for(int i=0;i<4;i++)
            {
                if(res.size() == m*n-1) {
                    res.add(matrix[x - 1][y - 1]);
                    return res;
                }
                while(visited[x+dx[i]][y+dy[i]])
                {
                    res.add(matrix[x-1][y-1]);
                    visited[x][y] = false;
                    x=x+dx[i];
                    y=y+dy[i];
                }
            }
        }
        return res;
    }
}
发布了241 篇原创文章 · 获赞 34 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/qq_38905818/article/details/104295288