1046. 打印方阵

题目描述

打印一个n*n的数字方阵，例如n=4时：(n<=100)
1 3 4 10
2 5 9 11
6 8 12 15
7 13 14 16

输入

输入n。

输出

输出n*n的方阵。

样例输入

4

样例输出

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

数据范围限制

C++代码

#include <iostream>

using namespace std;

const int N = 100;

int main()
{
    int Matrix[N][N];
    int n;
    cin >> n;

    int number = 1;
    int row, col;

    for(int k=2; k<=2*n; k++)
    {
        for(col=1; col<=n; col++)
        {
            for(row=1; row<=n; row++)
            {
                if (row+col == k)
                {
                    if (k>=4 && k%2 == 0)
                    {
                        Matrix[col-1][row-1] = number++;
                    }
                    else
                    {
                        Matrix[row-1][col-1] = number++;
                    }
                }
            }
        }
    }
    
    for(int row=1; row<=n; row++)
    {
        for(int col=1; col<=n; col++)
        {
            cout << Matrix[row-1][col-1] << " ";
        }
        cout << endl;
    }

    return 0;
}