螺旋矩阵
问题描述;
给你一个 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]
思路:用一个变量来表示往哪边走,按照右下左上的顺序开始走。然后当当前位置越界以及当前位置已经走过时调整位置。
代码:
class Solution {
public List<Integer> spiralOrder(int[][] matrix) {
List<Integer> ss = new ArrayList<Integer>();
int [][]flag = new int[matrix.length][matrix[0].length];
//定义四个方位
int to=0;
//0 : 右 1:下 2:左 3:上
int op=0;//列表的当前位置
int i=0,j=0;//从(0,0)开始
while(op <matrix.length*matrix[0].length){
//越界就换方向 下一次循环
if(i>=matrix.length || j>=matrix[0].length || i<0||j<0 ||flag[i][j] == 1) {
if(to == 0 ){
j--;
i++;
}
else if(to ==1 ){
i--;
j--;
}
else if(to ==2){
j++;
i--;
}
else if(to == 3){
i++;
j++;
}
to++;
to %=4;
continue;
}
if(to == 0) {
ss.add(matrix[i][j]);
flag[i][j]=1;
op++;
j++;
}
else if(to == 1){
ss.add(matrix[i][j]);
flag[i][j]=1;
op++;
i++;
}
else if(to == 2){
ss.add(matrix[i][j]);
flag[i][j]=1;
op++;
j--;
}
else if(to ==3){
ss.add(matrix[i][j]);
flag[i][j]=1;
op++;
i--;
}
}
return ss;
}
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/spiral-matrix