螺旋矩阵
- 题目
- 示例1
- 示例2
- 分析
- 代码
题目
一个 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]
分析
本题需要顺时针方向返回矩阵所有元素,相当于从外圈到内圈,每一圈都按照顺时针方向输出元素,直到所有元素都输出过为止。
还要注意上下左右四个方向访问的边界,哪些是开区间哪些是闭区间,代码如下。
代码
public List<Integer> spiralOrder(int[][] matrix) {
List<Integer> ans = new ArrayList<>();
int n = matrix.length, m = matrix[0].length;
int up = 0, down = n - 1, left = 0, right = m - 1;
int total = m * n;// 总的元素个数
while (ans.size() < total) {
// 上面,从左往右打印,左闭右闭区间[left,right]
for (int i = left; i <= right && ans.size() < total; i++)
ans.add(matrix[up][i]);
// 右边,从上往下打印,上开下开区间(up,down)
for (int i = up + 1; i <= down - 1 && ans.size() < total; i++)
ans.add(matrix[i][right]);
// 下边,从右往左打印,右闭左闭区间[right,left]
for (int i = right; i >= left && ans.size() < total; i--)
ans.add(matrix[down][i]);
// 左边,从下往上打印,下开上开区间(down,up)
for (int i = down - 1; i >= up + 1 && ans.size() < total; i--)
ans.add(matrix[i][left]);
left++;
right--;
up++;
down--;
}
return ans;
}