题目:输入一个矩阵,按照从外向里以顺时针的顺序依次打印出每一个数字。
示例:
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,2,3,6,9,8,7,4,5]
方法:按层模拟
这种比较好理解,一次遍历完一层元素
- 时间复杂度O(nm)
- 空间复杂度O(1)
public int[] spiralOrder(int[][] matrix) {
if (matrix.length == 0) {
return new int[0];
}
int row = matrix.length, col = matrix[0].length;
int[] res = new int[row * col];
int index = 0;
// up 为第一行, down 为最后一行,left 为第一列, right 为最后一列
int up = 0, down = row - 1, left = 0, right = col - 1;
while (true) {
// 顶行遍历
for (int i = left; i <= right; i++) {
res[index++] = matrix[up][i];
}
if (++up > down) {
break;
}
// 右列遍历
for (int i = up; i <= down; i++) {
res[index++] = matrix[i][right];
}
if (--right < left) {
break;
}
// 底行遍历
for (int i = right; i >= left; i--) {
res[index++] = matrix[down][i];
}
if (--down < up) {
break;
}
// 左列遍历
for (int i = down; i >= up; i--) {
res[index++] = matrix[i][left];
}
if (++left > right) {
break;
}
}
return res;
}