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

프로그래머스

 

문제 설명

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

입출력 예

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

 

코드

const solution = (n) => {
    const arr = Array(n);
    for(let i = 0; i < n; i++){
        arr[i] = [];
        for(let j = 0; j< n; j++){
            arr[i][j] = i === j ? 1 : 0;
        }
    }
    return arr;
}

 

설명

1) for 반복문을 사용하여 i는 행 j는 열로 초기화하여 i === j 일때 1을 할당하고 아니면 0을 할당합니다.