链接https://leetcode.cn/problems/spiral-matrix-ii/
题目:
给你一个正整数 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
方法:
思路
求解本题要坚持循环不变量原则。
模拟顺时针画矩阵的过程:
- 填充上行从左到右
- 填充右列从上到下
- 填充下行从右到左
- 填充左列从下到上
由外向内一圈一圈这么画下去。
这里一圈下来,我们要画每四条边,这四条边怎么画,每画一条边都要坚持一致的左闭右开,或者左开右闭的原则,这样这一圈才能按照统一的规则画下来。
这里每一种颜色,代表一条边,我们遍历的长度,可以看出每一个拐角处的处理规则,拐角处让给新的一条边来继续画。
这也是坚持了每条边左闭右开的原则。
先给n*n大小的矩阵申请空间,定义四个变量记录读取数字的起始位置,通过循环把数字放进数组,需要判断结束条件(left>right&&top>bottom),并判断区间开闭
遇到的问题:
①如何初始化m*n(n*n)的矩阵(数组)
//初始化返回数组
int**res=(int**)malloc(sizeof(int*)*n);
for(int i=0;i<n;i++)
{
res[i]=(int*)malloc(sizeof(int)*n);//res数组中每个元素大小都初始化为returnsize
memset(res[i],0,sizeof(int)*n);//通式 不打会越界
(*returnColumnSizes)[i]=n;//每一个元素都赋一个值,但赋值不可为0
}
②如何处理n=1的情况/如何判断区间开闭
代码
/**
* Return an array of arrays of size *returnSize.
* The sizes of the arrays are returned as *returnColumnSizes array.
* Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
*/
int** generateMatrix(int n, int* returnSize, int** returnColumnSizes){
//初始化返回数组长度大小
*returnSize=n;
*returnColumnSizes =(int*)malloc(sizeof(int)*n);
//初始化返回数组
int**res=(int**)malloc(sizeof(int*)*n);
for(int i=0;i<n;i++)
{
res[i]=(int*)malloc(sizeof(int)*n);//res数组中每个元素大小都初始化为returnsize
memset(res[i],0,sizeof(int)*n);//通式 不打会越界
(*returnColumnSizes)[i]=n;//每一个元素都赋一个值
}
int left=0,right=n-1,top=0,bottom=n-1;
int num=1;//表示存入数字
if(*returnSize==0)return res;
while(left<=right&&top<=bottom)
{
//注意n=1的情况 故要分开讨论
//n=1时左闭右闭 其余情况左闭右开/左开(+1)右闭
for(int j=left;j<=right;j++)
{
res[top][j]=num;
num++;
}
for(int j=top+1;j<=bottom;j++)
{
res[j][right]=num;
num++;
}
if(left<right&&top<bottom)//当n>1时
{
for(int j=right-1;j>left;j--)//注意j起始位置
{
res[bottom][j]=num;
num++;
}
for(int j=bottom;j>top;j--)//注意j起始位置
{
res[j][left]=num;
num++;
}
}
left++,right--,top++,bottom--;
}
return res;
}