HJ35 蛇形矩阵（c++）

题目链接：蛇形矩阵_牛客题霸_牛客网

这道题做的时候做错了，参考了评论区的方法

#include<iostream>
using namespace std;

int main()
{
    //输入行数
    int n;
    while(cin >> n){
    int num = 1;
    //总共输出n行
    for(int i = 1; i <= n; ++i){
        //输出每一行的第一个数字
        cout << num;
        int ans= num;
        //每一行的输出
        for(int j = i+1; j <= n; ++j){
            ans += j;
            cout <<" " <<ans;
        }
        //更新num值作为下一列的输出
        num += i;
        cout <<endl;
    }   
    }
    return 0;
}