用 JavaScript 構造一個 n 階單位矩陣

單位矩陣

單位矩陣是一個 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
   ]
]

AmitDiwan

更新日期：2020-12-11

750 次瀏覽

開啟您的 職業生涯

透過完成課程獲得認證

開始