题目
给你一个正整数 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
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/spiral-matrix-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题
模拟过程,其中的代码细节值得推敲;看完大神写的代码,再看看自己写的,只能说不堪入目。
这是自己第一次写的
int** generateMatrix(int n, int* returnSize, int** returnColumnSizes)
{
int** Matrix = malloc(n*sizeof(int*));
*returnSize = n;
*returnColumnSizes = malloc(n*sizeof(int));
int i,j;
for(i=0;i<n;i++){
Matrix[i] = malloc(n*sizeof(int));
(*returnColumnSizes)[i] = n; //开始写成*returnColumnSizes[i] = n,错误;注意运算优先级
}
for(i=0;i<n;i++){ //初始化为0,为了后面撞墙判断
for(j=0;j<n;j++){
Matrix[i][j] = 0;
}
}
int Tag = 1,K = 1;//Tag 控制转向,K为要赋的值
i = j = 0;
while(K <= n*n){
while(j>=0 && j<n){
if(Matrix[i][j]) break; //撞墙转向
Matrix[i][j] = K;
K++;
if(Tag == 1){
j++;
}else{
j--;
}
}
if(Tag == 1){
j--;
i++;
}else{
j++;
i--;
}
while(i>=0 && i<n){
if(Matrix[i][j]) break;
Matrix[i][j] = K;
K++;
if(Tag == 1){
i++;
}else{
i--;
}
}
if(Tag == 1){
i--;
j--;
}else{
i++;
j++;
}
Tag = -Tag;
}
return Matrix;
}
大神的优美代码
int** generateMatrix(int n, int* returnSize, int** returnColumnSizes)
{
int** Matrix = malloc(n*sizeof(int*));
*returnSize = n;
*returnColumnSizes = malloc(n*sizeof(int));
int i;
for(i=0;i<n;i++){
Matrix[i] = malloc(n*sizeof(int));
(*returnColumnSizes)[i] = n;
}
int Left = 0,Right = n-1,Top = 0,Bottom = n-1;
int K = 1;
while(K <= n*n){
for(i=Left;i<=Right;i++) Matrix[Top][i] = K++; //从左到右
Top++;
for(i=Top;i<=Bottom;i++) Matrix[i][Right] = K++; //从上到下
Right--;
for(i=Right;i>=Left;i--) Matrix[Bottom][i] = K++; //从右到左
Bottom--;
for(i=Bottom;i>=Top;i--) Matrix[i][Left] = K++; //从下到上
Left++;
}
return Matrix;
}
知识点总结
模拟过程中往往带有大量的判断,要注意细节, 尽可能的更多层次的抽象问题