[프로그래머스] 0단계 - 특별한 2차원 배열

1.  문제 설명

 

2. 내 답안

function solution(n) {
     // 먼저 n행인 1차원 배열 만들기
     let arr = new Array(n);
    
    // 반복문을 사용해 n행의 2차원 배열 만들기
     for (let i = 0; i < n; i++) {
        arr[i] = new Array(n);
         
         for (let j = 0; j < n; j++){
            arr[i][j] = i === j ? 1 : 0 
         }
    }
    
    return arr;
}

 

오랜만에 2차원 배열을 마주해서 살짝 시간이 걸리긴 했지만.. 그래도 수월하게 풀었다!

 

3. 참고할 만한 답안

function solution(n) {
  const answer = Array.from(Array(n), () => Array(n).fill(0));

  for (let i = 0; i < n; i++) {
    answer[i][i] = 1;
  }

  return answer;
}

 

Array.from은 처음 보는데 검색해보니까 아주 유용한 것 같아 기억하려고 적어둔다.