【LeetCode】#54螺旋矩阵(Spiral Matrix)
题目描述
给定一个包含 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]
Description
Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.
Example
Example 1:
Input:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
Output: [1,2,3,6,9,8,7,4,5]
Example 2:
Input:
[
[1, 2, 3, 4],
[5, 6, 7, 8],
[9,10,11,12]
]
Output: [1,2,3,4,8,12,11,10,9,5,6,7]
解法
class Solution {
public List<Integer> spiralOrder(int[][] matrix) {
if(matrix.length==0)
return new ArrayList<>();
int[] nums = new int[4];
nums[0] = matrix[0].length;
nums[1] = matrix.length;
nums[2] = -1;
nums[3] = 0;
int i=0, j=0, k=0;
List<Integer> res = new ArrayList<>();
while(true){
res.add(matrix[i][j]);
if(k==0){
if(j+1==nums[k]){
nums[k]--;
k = (k+1)%4;
if(i+1>=nums[k]){
break;
}else{
i++;
}
}else{
j++;
}
}else if(k==1){
if(i+1==nums[k]){
nums[k]--;
k = (k+1)%4;
if(j-1<=nums[k]){
break;
}else{
j--;
}
}else{
i++;
}
}else if(k==2){
if(j-1==nums[k]){
nums[k]++;
k = (k+1)%4;
if(i-1<=nums[k]){
break;
}else{
i--;
}
}else{
j--;
}
}else if(k==3){
if(i-1==nums[k]){
nums[k]++;
k = (k+1)%4;
if(j+1>=nums[k]){
break;
}else{
j++;
}
}else{
i--;
}
}
}
return res;
}
}