LeetCode 第59题:螺旋矩阵Ⅱ
LeetCode 第59题:螺旋矩阵Ⅱ
题目描述:
给你一个正整数n,生成一个包含1到n*n所有元素,且元素按顺时针螺旋排列的n*n正方形矩阵matrix。
示例1:
输入:n = 3 输出:[[1,2,3],[8,9,4],[7,6,5]]
示例2:
输入:n = 1 输出:[[1]]
提示:
1 <= n <= 20
解题思路:
初始化一个n*n的矩阵,元素均为0
- 用一个长为4的方向数组DIRS=[(0,1),(1,0),(0,-1),(-1,0)]分别表示右下左上四个方向。同时用一个下标di表示当前方向,初始值为0,表示一开始向右。
- 每次移动,相当于把行号增加DIRS[di][0],把列号增加DIRS[di][1]
- 向右转90°,相当于把di增加1,但在di=3时要回到di=0。两种情况合二为一,把di更新为(di+1)mod 4
const int DIRS[4][2]={{0,1},{1,0},{0,-1},{-1,0}};//右下左上 //整型指针returnSize,表示返回的二维数组的行数 //二级整型指针returnColumnSizes,表示返回的二维数组每一行的列数 int** generateMatrix(int n,int* returnSize,int** returnColumnSizes) {*returnSize = n;//二维数组有n行*returnColumnSizes = malloc(n*sizeof(int));//n列int **ans = malloc(n*sizeof(int*));for(int i=0;i<n;i++){(*returnColumnSizes)[i]=n;//二维数组每行元素为n个ans[i]=calloc(n,sizeof(int));//calloc 会设置分配的内存为零//分配n个元素大小为int的元素,且元素初始值为0}int i=0,j=0,di=0;for(int val=1;val<=n*n;val++){ans[i][j]=val;int x=i+DIRS[di][0],y=j+DIRS[di][1];//下一步位置//如果(x,y)出界或已经填入数字if(x<0||x>=n||y<=0||y>=n||ans[x][y])di=(di+1)%4;//右转90°i=i+DIRS[di][0],j=j+DIRS[di][1];//下一步}return ans; }