프로그래머스 181833 특별한 이차원 배열 1

1 개요

프로그래머스 181833 특별한 이차원 배열 1

• 프로그래머스 181833번 문제
• 프로그래머스 Lv. 0.6

2 C++

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(int n) {
    vector<vector<int>> answer;
    for(int i=0; i<n; i++) {
        vector<int> v;
        for(int j=0; j<n; j++) {
            v.emplace_back(i==j ? 1 : 0);
        }
        answer.emplace_back(v);
    }
    return answer;
}

#include <string>
#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] = 1;
    }
    return answer;
}