华为笔试题--蛇形矩阵

#include <iostream>
#include <vector>
using namespace std;

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

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