프로그래머스 JS - 특별한 이차원 배열 1, Lv.0

 

 

문제 설명

정수 n이 매개변수로 주어질 때, 다음과 같은 n × n 크기의 이차원 배열 arr를 return 하는 solution 함수를 작성해 주세요.
arr[i][j] (0 ≤ i, j < n)의 값은 i = j라면 1, 아니라면 0입니다.

 

 

 

 

문제 풀이

function solution(n) {
    var answer = [];
    for (i=0; i<n; i++) {
        answer.push(Array(n).fill(0));
    }
    for (i=0; i<answer.length; i++) {
        // answer 배열의 각 행에 대해서 반복
        const item = answer[i];
        // 현재 행을 나타내는 배열을 가져옴
        answer[i][i] = 1;
        // 현재 행의 대각선에 있는 요소를 1로 설정
    }

    return answer;
}