Xây dựng ma trận đơn vị cấp n trong JavaScript

Ma trận đơn vị (Identity Matrix)

Ma trận đơn vị là ma trận vuông n × n mà trên đường chéo chính gồm các 1 và các phần tử khác đều là 0.

Ví dụ ma trận đơn vị cấp 3 sẽ là:

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

Chúng ta cần viết một hàm JavaScript nhận vào một số, gọi là n, và trả về ma trận đơn vị cấp n×n.

Ví dụ

Dưới đây là mã nguồn:

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));

Kết quả

Đầu ra trên console:

[    [ 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    ] ]