核心思想:将二维数组每一圈分为4块,如图所示
class Solution {
public int[][] generateMatrix(int n) {
int startx=0;//竖行开始位置
int starty=0;//横行开始位置
int offset=1;//结束偏移量
int[][] nums=new int[n][n];
int i,j;
int num=1;//数组初始值
int count=n/2;//绕几圈
while(count>=0){
for(j=starty;j<n-offset;j++)nums[startx][j]=num++;//黄色
for(i=startx;i<n-offset;i++)nums[i][j]=num++;//粉色
for(;j>starty;j--)nums[i][j]=num++;//蓝色
for(;i>startx;i--)nums[i][j]=num++;//紫色
startx++;//若有第二圈,竖行开始位置+1
starty++;//若有第二圈,横行开始位置+1
offset+=1;//偏移量+1
count--;//循环次数-1
}
if(n%2==1)nums[n/2][n/2]=num;//若n为奇数,则最中间的不会被循环,需要另外赋值
return nums;
}
}