JustOJ1500: 蛇行矩阵

题目链接：https://oj.ismdeep.com/problem?id=1500

题目描述

蛇形矩阵是由1开始的自然数依次排列成的一个矩阵上三角形。

输入

本题有多组数据，每组数据由一个正整数N组成。（N不大于100）

输出

对于每一组数据，输出一个N行的蛇形矩阵。两组输出之间不要额外的空行。矩阵三角中同一行的数字用一个空格分开。行尾不要多余的空格。

样例输入

5

样例输出

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

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
int a[100][100];
int main()
{
    int n;
    while(cin>>n){
        for(int i=0;i<n;i++){
            for(int j=0;j<n-i;j++){
                if(i==0&&j==0) a[i][j]=1;
                else if(i==0&&j>0) a[i][j]=a[i][j-1]+j+1;
                else if(i>0&&j==0) a[i][j]=a[i-1][j]+i;
                else if(i&&j) a[i][j]=a[i][j-1]+i+j+1;
            }
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<n-i;j++){
                if(j==0) cout<<a[i][j];
                else cout<<" "<<a[i][j];
                
            }
            cout<<endl;
        }
    }
     return 0;
}