JavaScript에서 n차의 단위 행렬 구성

Identity Matrix

항등 행렬은 대각선이 1로 구성되고 다른 요소가 모두 0인 n × n 정방 행렬인 행렬입니다.

예를 들어 순서의 단위 행렬은 다음과 같습니다 -

const arr = [
   [1, 0, 0],
   [0, 1, 0],
   [0, 0, 1]
];

n과 같은 숫자를 받아서 n*n 차수의 단위 행렬을 반환하는 JavaScript 함수를 작성해야 합니다.

예시

다음은 코드입니다 -

const num = 5;
const constructIdentity = (num = 1) => {
   const res = [];
   for(let i = 0; i < num; i++){
      if(!res[i]){
         res[i] = [];
      };
      for(let j = 0; j < num; j++){
         if(i === j){
            res[i][j] = 1;
         }else{
            res[i][j] = 0;
         };
      };
   };
   return res;
};
console.log(constructIdentity(num));

출력

다음은 콘솔의 출력입니다 -

[
   [ 1, 0, 0, 0, 0 ],
   [ 0, 1, 0, 0, 0 ],
   [ 0, 0, 1, 0, 0 ],
   [ 0, 0, 0, 1, 0 ],
   [
      0,
      0,
      0,
      0,
      1
   ]
]