C语言蛇形矩阵2

#include<stdio.h>
#include<malloc.h>
int main()
{
int **p=NULL;
int n,i,j,k=1,q=0,line;
printf("请输入矩阵的行数:");
scanf("%d",&n);
if((p=(int**)malloc(sizeof(int**)*n))==NULL)//根据行数分配内存，分配双指针
{
printf("不能分配内存!!!\n");
return 1;
}
for(i=0;i<n;i++)
{
if((p[i]=(int*)malloc(sizeof(int*)*n))==NULL)//为每一行分配指针，每一行有n个数
{
printf("不能分配内存!!!\n");
return 1;
}
}
for(i=0,j=0;k<=n*n;q++)//q是斜线上i+j相等的值,例如p[0][2],p[1][1],p[2][0]中q的值为2；
{
if(q%2==0)//如果q是偶数，就从下往上开始赋值
{
for(line=i;line>=0;line--)
{
if(q-line<n)//防止数组越界，比如声明4*4的数组当i=0,q=5时,q-line为5,越界了
p[line][q-line]=k++;
}
}
else//如果q是奇数，就从上往下开始赋值
{
for(line=0;line<=i;line++)
{
if(q-line<n)
p[line][q-line]=k++;
}
}
if(i<n-1)
{
i++;//防止数组越界i,j的值最大为n-1,而q的值最大为2*(n-1)
j++;
}
}
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
{
printf("%-6d",p[i][j]);
}
printf("\n");
}
for(i=0;i<n;i++)//分别释放每一行的指针
{
free(p[i]);
p[i]=NULL;
}
free(p);//释放双指针
p=NULL;
return 0;
}