在 JavaScript 中构建 n 阶单位矩阵

javascriptweb developmentfront end technologyobject oriented programming更新于 2024/2/22 9:16:17

单位矩阵

单位矩阵是一个 n × n 的方阵,对角线由 1 组成,其他元素均为 0。

例如 阶单位矩阵为 −

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

我们需要编写一个 JavaScript 函数,该函数接受一个数字(例如 n),并返回 n*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));

输出

以下是控制台上的输出 −

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

相关文章


有用资源