思路
先把数组都置为零,然后一个个填数
题目
给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
示例 1:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
提示:
1 <= n <= 20
解题方法
先把所有的数都置为零,然后按照循环的顺序依次填入数字。
以向右为例:当加一之后没有超出数组的边界并且该位置的数为0,则证明可以继续填数。一旦加一超出数组边界或者已经填写过,就开始考虑向下,然后再依次考虑向左、向上
代码
class Solution {
public int[][] generateMatrix(int n) {
/*把所有的与元素都设置为0,如果右侧的数为0就往右走,如果下面的数为0就往下走,向左向上也类似*/
int[][] a = new int[n][n];
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
a[i][j] = 0;
}
}
int i = 0;//表示行
int j = 0;//表示列
int num = 1;//用来计数,填写矩阵
a[0][0] = num++;//先把第一个位置赋值,因为有可能n=1,后边直接不用管了
if(n==1){//直接返回就可以了
return a;
}
while (true) {
//向右
while ((j + 1) < n && a[i][j + 1] == 0) {
j++;
a[i][j] = num++;
}
//向下
while ((i + 1) < n && a[i+1][j] == 0) {
i++;
a[i][j] = num++;
}
//向左
while ((j - 1) >= 0 && a[i][j - 1] == 0) {
j--;
a[i][j] = num++;
}
//向上
while ((i - 1) >= 0 && a[i-1][j] == 0) {
i--;
a[i][j] = num++;
}
//如果num==(n*n+1)则证明已经把所有的位置都填上数了
//因为这里采用的是num++而不是++num,所以是num==(n*n+1)而不是num==n*n
if(num==(n*n+1)){
return a;
}
}
}
}