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;
}