题目:
Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.
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]
解答
题目要求为顺时针返回矩阵的所有元素
解题思路为:
分别设定 top、down、left、right 四个边界值,分别指向剩余待遍历的矩阵的上边界、下边界、左边界和右边界。从矩阵左上角第一个元素开始,分别沿右、下、左、上四个方向走,当达到边界条件时,则更改为下一个方向,并缩短对应的边界值
class Solution {
public List<Integer> spiralOrder(int[][] matrix) {
List<Integer> res = new ArrayList<>();
if(matrix.length == 0) {
return res;
}
int row = matrix.length; //行数
int colnmn = matrix[0].length; //列数
int top = 0;
int left = 0;
int right = colnmn-1;
int down = row-1;
while(res.size() < row * colnmn) {
for(int i=left; i<=right; i++) {
res.add(matrix[top][i]);
}
top++;
if(res.size() < row * colnmn) {
for(int i=top; i<=down; i++) {
res.add(matrix[i][right]);
}
right--;
}
if(res.size() < row * colnmn) {
for(int i=right; i>=left; i--) {
res.add(matrix[down][i]);
}
down--;
}
if(res.size() < row * colnmn) {
for(int i=down; i>=top; i--) {
res.add(matrix[i][left]);
}
left++;
}
}
return res;
}
}
注意对if(matrix.length == 0)
的判空处理,第一次提交忽略导致错误。