蛇行矩阵 （C++代码）

参考代码:

#include <iostream>
using namespace std;
int main()
{
    int n;
    while(cin>>n){
        int head=1;
        for(int i=0;i<n;i++){
            head+=i;
            int dig=head;
            cout<<head;
            for(int j=1;j<n-i;j++){
                dig+=j+i+1;
                cout<<" "<<dig;
            }
            cout<<endl;
        }
    }
    return 0;
}