题解 | #蛇形矩阵#

// 抄评论区里面的代码写的。
#include<iostream>
#include <vector>

using namespace std;

int main(){
    int n;
    cin >> n;
    int num = 1;
    vector<vector<int>> matrix(n, vector(n, 0));
    for(int i = 0; i < n; i++){
        int j = i;
        int k = 0;
        while(j >= 0){
            matrix[j][k] = num;
            num++;
            j--;
            k++;
        }
    }

    for(int i = 0; i < n; i++){
        for(int j = 0; j < n; j++){
            if(matrix[i][j] != 0){
                cout << matrix[i][j] << " ";
            }
        }
        cout << endl;

    }

    return 0;
}