Level 0. 특별한 이차원 배열 1

Level 0. 특별한 이차원 배열 1

 

 

#include <iostream>
#include <vector>

using namespace std;

vector<vector<int>> solution(int n)
{
    vector<vector<int>> answer(n, vector<int>(n, 0));

    for (int i = 0; i < n; ++i)
        ++answer[i][i];

    return answer;
}