给你一个 m 行 n 列的矩阵 matrix ,请按照 顺时针螺旋顺序 ,返回矩阵中的所有元素。
示例 1:
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,2,3,6,9,8,7,4,5]
示例 2:
输入:matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
输出:[1,2,3,4,8,12,11,10,9,5,6,7]
思路:
可以模拟螺旋矩阵的路径。初始位置是矩阵的左上角,初始方向是向右,当路径超出界限或者进入之前访问过的位置时,顺时针旋转,进入下一个方向。(给定一个二维数组,表示右下左上四个方向,再给一个变量来控制方向)
判断路径是否进入之前访问过的位置需要使用一个与输入矩阵大小相同的辅助矩阵 vis,其中的每个元素表示该位置是否被访问过。当一个元素被访问时,将 vis中的对应位置的元素设为已访问。
如何判断路径是否结束?由于矩阵中的每个元素都被访问一次,因此路径的长度即为矩阵中的元素数量,当路径的长度达到矩阵中的元素数量时即为完整路径,将该路径返回。(求出矩阵的长和宽,长*宽即为总元素个数)
题解
class Solution {
public List<Integer> spiralOrder(int[][] matrix) {
List<Integer> order = new ArrayList<>();
if(matrix==null)
return order;
int row = matrix.length,col = matrix[0].length;
boolean[][] vis = new boolean[row][col];
int total = row*col;
int x=0,y=0;
int[][] dir = {{0,1},{1,0},{0,-1},{-1,0}}; //顺时针走
int dirIndex = 0; //控制方向
for(int i=0;i<total;i++) //有多少元素,就走多少次
{
order.add(matrix[x][y]);
vis[x][y] = true; //走过就标记为true
int tx = x+dir[dirIndex][0],ty = y+dir[dirIndex][1];//下一个坐标,作为临时坐标,用来进行下面的判断
if(tx<0||tx>=row||ty<0||ty>=col||vis[tx][ty]) //越界或者已经走过,就改变方向
{
dirIndex = (dirIndex+1)%4;
}
x += dir[dirIndex][0];//更新
y += dir[dirIndex][1];
}
return order;
}
}