蛇形矩阵

#include<stdio.h>
#define max 20
int main()
{
int i=0,j=0;
int k;
int n;
int a[max][max];
printf("请输入矩阵的首数字:\n");
scanf("%d",&k);
printf("请输入矩阵的阶数:\n");
scanf("%d",&n);
while(i<n && j<n)
{
while(j<n && i>=0)
{
a[j][i]=k++;
j++;
i--;
}
if(i<0 && j<n)
i=0;
else
{
j=n-1;
i+=2;
}
while(j>=0 && i<n)
{
a[j][i]=k++;
j--;
i++;
}
if(j<0 && i<n)
j=0;
else
{
j+=2;
i=n-1;
}
}
printf("请输出蛇形矩阵:\n");
for(i=0;i<n;i++)
for(j=0;j<n;j++)
{
if(j%n==0)
printf("\n");
printf("%4d",a[i][j]);
}
return 0;
}