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

◈ 문제 설명

◈ 문제 설명 링크
코딩테스트 연습 - 특별한 이차원 배열 1 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

◈ 작성 코드

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