Сформировать матрицу по заданному образцу

#include <iostream>
#include <vector>
#include <iomanip>
 
int main()
{
    size_t const n = 3;
    std::vector<std::vector<int> > matrix(n);
 
    for(size_t i = 0; i < n; ++i)
        for(size_t j = 0; j < n; ++j)
            matrix.at(i).push_back(i == j ? i : 0);
 
    for(size_t i = 0; i < n; ++i)
    {
        for(size_t j = 0; j < n; ++j)
            std::cout << std::setw(8) << std::left << matrix.at(i).at(j);
        std::cout << '\n';
    }
    return 0;
}

#include <iostream>
#include <conio.h>
using namespace std;
int main() {
    int N;
    cin >> N;
    int mas[N][N];
    for(int i=0; i<N; i++) {
            for(int j=0; j<N; j++) {
                    mas[i][j] = 0;
            }
    }
    for(int i=0; i<N; i++) {
            for(int j=0; j<N; j++) {
                    mas[i][j] = j;
                    ++i;
            }
    }
    for(int i=0; i<N; i++) {
            for(int j=0; j<N; j++) {
                    cout << mas[i][j] << " ";
            }
            cout << endl;
    }
    getche();
}