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

문제

특별한 이차원 배열 1 : 문제 링크

문제 분석

• 정수 n이 매개변수로 주어질 때, 다음과 같은 n x n 크기의 이차원 배열 arr를 return

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

• n x n 크기의 이차원 배열 answer의 원소를 0으로 초기화. 이중 for loop를 통해 이차원 배열의 모든 원소를 검사하고, i = j 조건을 만족할 경우 해당 위치의 원소를 1로 저장. 최종적으로 저장된 answer를 return

풀이

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