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

문제 설명

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

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

 

제한사항

• 1 ≤ n ≤ 100

 

입출력 예

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

 

나의 풀이

function solution(n) {
    const answer = [];
    let arr =[]
    for(let i=0; i<n; i++){
        for(let j=0; j<n; j++){
            if(i===j){
                arr.push(1)
            }else{
                arr.push(0)
            }
        }
        answer.push(arr) // arr = [1, 0, 0], arr =[0, 1, 0],  arr =[0, 0, 1]
        arr=[] // 초기화
    }
    return answer;
}