n*n蛇形矩阵

问题描述：

蛇形矩阵是一个nn的矩阵，将整数1到nn按照蛇形的顺序装入一个 n*n 的蛇形矩阵中，如样例所示分别为5阶和10阶蛇形矩阵：

输入与输出要求：

输入一个整数n，代表蛇形矩阵的阶数，n的范围是1—100。输出蛇形矩阵。每行的每个元素用空格分隔，注意最后一个数的后面为换行符。

程序运行效果：

Sample 1：
5

1 3 4 10 11
2 5 9 12 19
6 8 13 18 20
7 14 17 21 24
15 16 22 23 25
Sample 2：
10

1 3 4 10 11 21 22 36 37 55
2 5 9 12 20 23 35 38 54 56
6 8 13 19 24 34 39 53 57 72
7 14 18 25 33 40 52 58 71 73
15 17 26 32 41 51 59 70 74 85
16 27 31 42 50 60 69 75 84 86
28 30 43 49 61 68 76 83 87 94
29 44 48 62 67 77 82 88 93 95
45 47 63 66 78 81 89 92 96 99
46 64 65 79 80 90 91 97 98 100

代码实现：

#include<stdio.h>
#include<stdlib.h>
int main()
{
	int n,i=0,j=0,k=1;
	scanf("%d",&n);
	int **a;
	a=(int**)malloc(sizeof(int*)*n);
	for(i=0;i<n;i++)
	{
		a[i]=(int*)malloc(sizeof(int)*n);
		for(j=0;j<n;j++)
		{
			a[i][j]=1;
		}
	}
	i=0,j=0;
	//定义四种状态：k=1向下走，k=2向右上方走，k=3向右走，k=4向左下角走
	while(i!=n-1||j!=n-1)
	{
		if(k==1)
		{
			a[i+1][j]=a[i][j]+1;
			i++;
			if(j==0)         k=2;
			else if(j==n-1)  k=4;
		}
		
		else if(k==2)
		{
			a[i-1][j+1]=a[i][j]+1;
			i--,j++;
			if(i==0&&j!=n-1) k=3;
			else if(j==n-1)  k=1;
			else             k=2;
		}
		
		else if(k==3)
		{
			a[i][j+1]=a[i][j]+1;
			j++;
			if(i==0)         k=4;
			else if(i==n-1)  k=2;
		}
		
		else if(k==4)
		{
			a[i+1][j-1]=a[i][j]+1;
			i++,j--;
			if(j==0&&i!=n-1) k=1;
			else if(i==n-1)  k=3;
			else             k=4; 
		}
	}
	
	for(i=0;i<n;i++)
	{
		for(j=0;j<n-1;j++)
		  printf("%d ",a[i][j]);
		printf("%d\n",a[i][n-1]);
	}
	
	for(i=0;i<n;i++)
	{
		free(a[i]);
	}
	free(a);
	a=NULL;
	return 0;
}