蛇形填数和蛇形取数（基础模拟练习）

/*
问题 输入矩阵的规模n,先将数按照下，右，上，左的顺序填入矩阵，再按照这样的顺序取出。
解题思路 模拟，按照笔的顺序存入取出，注意初始化的时候一定将矩阵全部初始化。 
*/ 
#include<cstdio>
#include<cstring>
const int N=1001;
int a[N][N],b[N][N];
int main()
{
    int n,i,j,tot,x,y;
    while(scanf("%d",&n) != EOF){
        //蛇形填数 
        x=0;
        y=0;
        tot=0;
        memset(a,0,sizeof(int)*N*N);//一定要初始化全部,否则出现不可预估的错误 
        while(tot < n*n){
            while(x < n && !a[x][y]){
                a[x][y] = ++tot;
                x++;
            }
            
            x--;
            y++;
            while(y < n && !a[x][y]){
                a[x][y]=++tot;
                y++;
            }
            
            y--;
            x--;
            while(x >= 0 && !a[x][y]){
                a[x][y] = ++tot;
                x--;
            }
            
            x++;
            y--;
            while(y >= 0 && !a[x][y]){
                a[x][y] = ++tot;
                y--;
            }
            
            y++;
            x++;
        }
        for(i=0;i<n;i++){
            for(j=0;j<n;j++){
                printf("%d ",a[i][j]);
            }
            printf("
");
        }
        //蛇形取数 
        x=0;
        y=0;
        tot=0;
        memset(b,0,sizeof(int)*N*N);
        while(tot < n*n){
            while(x < n && !b[x][y]){
                printf("%d ",a[x][y]);
                b[x][y]=1;
                ++tot;
                x++;
            }
            x--;
            y++;
            while(y < n && !b[x][y]){
                printf("%d ",a[x][y]);
                b[x][y]=1;
                ++tot;
                y++;
            }
            
            y--;
            x--;
            while(x >= 0 && !b[x][y]){
                printf("%d ",a[x][y]);
                b[x][y]=1;
                ++tot;
                x--;
            }
            
            x++;
            y--;
            while(y >= 0 && !b[x][y]){
                printf("%d ",a[x][y]);
                b[x][y]=1;
                ++tot;
                y--;
            }
            
            y++;
            x++;
        }
        printf("
");
    }
    return 0;
}