今天学习数组系列算法刷到这题,Leetcode描述如下:
此题应该算纯数组题中比较经典难度较高的题。根据题目描述我们要将1到n*n数字存入一个nxn的矩阵。
首先很容易根据得出思路,我们每次在矩阵最外围的一圈将数字存入,然后将矩阵缩小再次循环。
int[][] box = new int[n][n]; //矩阵
int value = 1; //记录填入数字
int min_step = 0; //循环的左上界
int max_step = n-1; //循环的右下界
而为了正好能一圈填满,我们不能一次将一条边填满,因此我们采用左闭右开区间填写数字。 每次填写完一圈我们将更新上下界并将指针指向该圈左上角。注释给大家做出了每一步的解读。
//循环保证将1到n*n数字填写完退出循环
while(value<=n*n){
//将位置设为左上角
int x = min_step;
int y = min_step;
//若为n为奇数,将中心填写为最后一个value,也就是n*n
if(min_step==max_step){
box[x][y] = value++;
}
//上边
while(y<max_step){
box[x][y] = value++;
y++;
}
//右边
while(x<max_step){
box[x][y] = value++;
x++;
}
//下边
while(y>min_step){
box[x][y] = value++;
y--;
}
//左边
while(x>min_step){
box[x][y] = value;
x--;
}
//进入内一圈循环
min_step++;
max_step--;
}
下面是完整代码,我认为比官解更易读一点。仅供大家参考,不喜勿喷~~
class Solution {
public int[][] generateMatrix(int n) {
int[][] box = new int[n][n];
int value = 1;
int min_step = 0;
int max_step = n-1;
while(value<=n*n){
int x = min_step;
int y = min_step;
if(min_step==max_step){
box[x][y] = value++;
}
while(y<max_step){
box[x][y] = value++;
y++;
}
while(x<max_step){
box[x][y] = value++;
x++;
}
while(y>min_step){
box[x][y] = value++;
y--;
}
while(x>min_step){
box[x][y] = value++;
x--;
}
min_step++;
max_step--;
}
return box;
}
}