[C++] 프로그래머스 특별한 이차원 배열 1

문제설명

정수 n이 매개변수로 주어질 때, 다음과 같은 n × n 크기의 이차원 배열 arr를
return 하는 solution 함수를 작성해 주세요.

• arr[i][j] (0 ≤ i, j < n)의 값은 i = j라면 1, 아니라면 0입니다.

제한사항

• 1 ≤ n ≤ 100

입출력 예

n	result
3	[[1, 0, 0], [0, 1, 0], [0, 0, 1]]
6	[[1, 0, 0, 0, 0, 0], [0, 1, 0, 0, 0, 0], [0, 0, 1, 0, 0, 0], [0, 0, 0, 1, 0, 0], [0, 0, 0, 0, 1, 0], [0, 0, 0, 0, 0, 1]]
1	[[1]]

풀이

문제의 조건에 맞게 n x n 크기의 이차원 배열을 만드는 구현 문제입니다.

#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> temp;
        
        for(int j = 0; j < n; j++)
        {
            if(j == i)
                temp.push_back(1);
            else
                temp.push_back(0);
        }
        
        answer.push_back(temp);
    }
    
    return answer;
}

결과