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