给定一个包含 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]
public List<Integer> spiralOrder(int[][] matrix) {
List<Integer> result = new ArrayList<>();
int height = matrix.length, width = height == 0 ? 0 : matrix[0].length;
int size = height * width;
int[] dirX = {0, 1, 0, -1};
int[] dirY = {1, 0, -1, 0};
// 初始化起点坐标:(0,-1) 方向:向右;
int x = 0, y = -1, dir = 0;
for (int step, total = 0; total < size; total += step) {
// 根据方向得到对应的位移, 并修正此后矩阵的参数(此后线段的长度)
if (dir == 0 || dir == 2) {
step = width;
height--;
} else {
step = height;
width--;
}
// 此前确定了起点坐标、方向和位移, 就可以得到当前线段的所有坐标,并输出到结果集;
for (int i = step; i > 0; i--) {
x += dirX[dir];
y += dirY[dir];
result.add(matrix[x][y]);
}
// 调整下一条线段方向
dir = ++dir % 4;
}
return result;