信息學奧賽C++語言： 蛇形方陣1

【題目描述】
輸入一個正整數 n，生成一個 n×n 的蛇形方陣（具體見樣例）。

【輸入】
一行一個正整數 n，1≤n≤20。

【輸出】
共 n 行，每行 n 個正整數，每個正整數佔 5 列。

【輸入樣例】
5

【輸出樣例】

代碼

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int n,j,i,k,t=0,a[21][21];
	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;
}