54. 螺旋矩阵
螺旋矩阵
给你一个 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]
思路
- 边界控制:我们使用四个变量来控制当前遍历的边界:
up
: 当前遍历的最上面一行。down
: 当前遍历的最下面一行。left
: 当前遍历的最左边一列。right
: 当前遍历的最右边一列。
- 螺旋遍历:我们按顺时针方向,从左到右、从上到下、从右到左、从下到上 依次遍历当前边界范围内的矩阵元素,并更新边界值。直到所有元素都被遍历。
class Solution {
public List<Integer> spiralOrder(int[][] matrix) {
List<Integer> res = new ArrayList<>();
int m = matrix.length;
int n = matrix[0].length;
int up = 0, down = m - 1, left = 0, right = n - 1;
while(true){
//从左到右遍历上面第up行
for(int i = left; i <= right; i++){
res.add(matrix[up][i]);
}
if(++up > down){
break;
}
//从上到下遍历右边第right列
for(int i = up; i <= down; i++){
res.add(matrix[i][right]);
}
if(--right < left){
break;
}
//从右到左遍历下面第down行
for(int i = right; i >= left; i--){
res.add(matrix[down][i]);
}
if(--down < up){
break;
}
//从下到上遍历左边第left列
for(int i = down; i >= up; i--){
res.add(matrix[i][left]);
}
if(++left > right){
break;
}
}
return res;
}
}