思路:此题思路就是绕圈遍历,全靠条件处理技巧,重点要清楚的就是循环不变量:左闭右开(即拐弯处的一个数,留给第二行处理)
以下是代码随想录的作者的一张图片,每次for循环,只遍历相同颜色的元素,就是此题的左闭右开原则。
class Solution {
public int[][] generateMatrix(int n) {
int [][] ans = new int [n][n];
int loop = 0; //while循环次数
int curNum=1; //当前赋值大小
int start = 0; //每圈开始(行,列)的起始位置,如(1,1)
int i=0; //当前遍历到的行位置
int j=0; //当前遍历到的列位置
//循环不变量:左闭右开(即拐弯处的一个数,留给第二行处理)
//loop是while循环次数,判断完边界后,loop+1,加1后的loop作为每行后半截不要遍历的个数
while(loop++ < n/2){
//1、从左到右遍历,结束时,j是多加了1的,刚好作为下一个的列
//每轮开始,i,j都要重新赋值初始值为start,因为上一轮结束后,i,j变为0了,而第二轮,是从1开始
for(j=start; j<n-loop;j++ ){
ans[start][j] = curNum++;
}
//2、从上到下遍历,j刚好用上面加完的
//每轮开始,i,j都要重新赋值初始值为start,因为上一轮结束后,i,j变为0了,而第二轮,是从1开始
for(i=start; i<n-loop; i++) {//每轮开始,i,j都要重新赋值初始值为start
ans[i][j] = curNum++;
}
//3、从右到左,此时j的初始值就是j本身,所以for循环 j没有赋初始值
for(; j>start;j--){
ans[i][j] = curNum++;
}
//4、从下到上,此时i的初始值也是i本身,同理
for(; i>start; i--){
ans[i][j] = curNum++;
}
start++;
}
//n是奇数时,剩最中间的一个值赋值
if(n%2==1) {
ans[start][start]=curNum;
}
return ans;
}
}