54. 螺旋矩阵
给你一个 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]
注:matrix可以理解为二维数组
代码:
public List<Integer> spiralOrder(int[][] matrix) {
List<Integer> list = new ArrayList<>();
int iMin = 0, iMax = matrix.length - 1;
int jMin = 0, jMax = matrix[iMin].length - 1;
int i = 0, j = 0;
while (iMin <= iMax && jMin <= jMax) {
if (iMin == iMax) {
for (j = jMin; j <= jMax; j++) {
list.add(matrix[iMin][j]);
}
} else if (jMin == jMax) {
for (i = iMin; i <= iMax; i++) {
list.add(matrix[i][jMin]);
}
} else {
// left to right
for (j = jMin; j < jMax; j++) {
list.add(matrix[iMin][j]);
}
// up to down
for (i = iMin; i < iMax; i++) {
list.add(matrix[i][jMax]);
}
// right to left
for (j = jMax; j > jMin; j--) {
list.add(matrix[iMax][j]);
}
// down to up
for (i = iMax; i > iMin; i--) {
list.add(matrix[i][jMin]);
}
}
iMin++;
iMax--;
jMin++;
jMax--;
}
return list;
}
59. 螺旋矩阵 II
给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
示例 1:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
代码:
public int[][] generateMatrix(int n) {
int[][] matrix = new int[n][n];
int num = 1;
int i = 0, j = 0;
int iMin = 0, iMax = n - 1;
int jMin = 0, jMax = n - 1;
while (iMin <= iMax && jMin <= jMax) {
if (iMin == iMax) {
for (j = jMin; j <= jMax; j++) {
matrix[iMin][j] = num;
}
}else if (jMin == jMax) {
for (i = iMin; i <= iMax; i++) {
matrix[i][jMin] = num;
}
} else {
// left to right
for (j = jMin; j < jMax; j++) {
matrix[iMin][j] = num;
num++;
}
// up to down
for (i = iMin; i < iMax; i++) {
matrix[i][jMax] = num;
num++;
}
// right to left
for (j = jMax; j > jMin; j--) {
matrix[iMax][j] = num;
num++;
}
// down to up
for (i = iMax; i > iMin; i--) {
matrix[i][jMin] = num;
num++;
}
}
iMin++;
iMax--;
jMin++;
jMax--;
}
return matrix;
}