问题 : 蛇行矩阵

题目描述
蛇形矩阵是由 1 开始的自然数依次排列成的一个矩阵上三角形
输入
正整数 N表示层数，N 不大于 100
输出
输出一个 N 行的蛇形矩阵，矩阵三角中同一行的数字用一个空格分开，行尾不要多余的空格。
样例输入
5
样例输出
1 3 6 10 15
2 5 9 14
4 8 13
7 12
11

AC代码

#include <iostream>
#include<cstring>
using namespace std;
int a[100][100];
int main()
{
    int n,p=1;
    cin>>n;
    memset(a,0,sizeof(a));
    for(int i=0;i<n;i++){
        int temp=i,j=0;
        while(temp>=0){
            a[temp][j]=p;
            p++;
            temp--;j++;
        }
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(a[i][j]!=0){
                 if(j==0)
                    cout<<a[i][j];
                else
                    cout<<" "<<a[i][j];
            }else{
                break;
            }
        }
        cout<<endl;
    }
    return 0;
}