信息奥赛课课通（C++）蛇形方阵

题目描述

输入一个正整数 n，生成一个 n×n 的蛇形方阵（具体见样例）。

输入

一行一个正整数 n，1≤n≤20。

输出

共 n 行，每行 n 个正整数，每个正整数占 5 列。

样例输入 Copy

5

样例输出 Copy

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

代码

#include<bits/stdc++.h>
using namespace std;
int n,j,i,k,t=0,a[999][999];
int main()
{
    cin>>n;
    for(k=1;k<=n;k++)
        if(k%2)
            for(j=1;j<=k;j++){
                i=k+1-j;
                t++;
                a[i][j]=t;
                a[n+1-i][n+1-j]=n*n+1-t;
            }
        else
            for(j=k;j>=1;j--){
                i=k+1-j;
                t++;
                a[i][j]=t;
                a[n+1-i][n+1-j]=n*n+1-t;
            }
    for(i=1;i<=n;i++){
        for(j=1;j<=n;j++)
        cout<<setw(5)<<a[i][j];
        cout<<endl;
    }
    return 0;
}