题目
给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出: [
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ] ]
解题思路
从矩形的外部一直往内部螺旋的加入数字就好了,类似于之前的螺旋矩阵1的解法,但是这次是正方形,所以这题的下标更好理解。
代码如下
class Solution {
public int[][] generateMatrix(int n) {
int[][] res = new int[n][n];
//代表数组长度res.length-1;
n--;
int num = 1;
for(int i = 0; i <= n/2; i ++) {
//剩下最后一个的情况
if(n == 2*i) {
res[i][i] = num;
break;
};
num = addRow(res, i, i, n-i, num);
num = addCol(res, n-i, i, n-i, num);
num = addRow(res, n-i, n-i, i, num);
num = addCol(res, i, n-i,i, num);
}
return res;
}
//返回值保存num的值
public int addRow(int[][] res,int row, int colStart, int colEnd, int num) {
if(colStart == colEnd) {
res[row][colStart] = num++;
return num;
}
for(int i = colStart;i < colEnd; i ++) res[row][i] = num++;
for(int i = colStart;i > colEnd; i --) res[row][i] = num++;
return num;
}
public int addCol(int[][] res,int col, int rowStart, int rowEnd, int num) {
if(rowStart == rowEnd) {
res[col][rowEnd] = num++;
return num;
}
for(int i = rowStart;i < rowEnd; i ++) res[i][col] = num++;
for(int i = rowStart;i > rowEnd; i --) res[i][col] = num++;
return num;
}
}
提交结果
成功
显示详情
执行用时 : 0 ms, 在Spiral Matrix II的Java提交中击败了100.00% 的用户
内存消耗 : 33.7 MB, 在Spiral Matrix II的Java提交中击败了84.01% 的用户