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

---

문제설명

 

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

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

---

제한사항

• 1 ≤ n ≤ 100

---

입출력 예

---

작성코드

#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;
}

---

문제 URL

https://school.programmers.co.kr/learn/courses/30/lessons/181833

프로그래머스

---