WZOI-394蛇形填充数组

题目描述:

用数字1,2,3,4,...,n*n这n2个数蛇形填充规模为n*n的方阵。

蛇形填充方法为：

对于每一条左下-右上的斜线，从左上到右下依次编号1,2,...,2n-1；按编号从小到大的顺序，将数字从小到大填入各条斜线，其中编号为奇数的从左下向右上填写，编号为偶数的从右上到左下填写。

比如n=4时，方阵填充为如下形式：

1  2  6  7
3  5  8  13
4  9  12 14
10 11 15 16

输入格式:

输入一个不大于10的正整数n，表示方阵的行数。

输出格式:

输出该方阵，相邻两个元素之间用单个空格间隔。

样例输入:

4

样例输出:

1 2 6 7
3 5 8 13
4 9 12 14
10 11 15 16

时间限制: 1000ms
空间限制: 128MB

代码如下：

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