JavaScript 求矩形对角线之和 矩形的宽由用户输入

/* 
求矩形对角线之和 矩形的宽由用户输入

请输入矩阵的宽
3
[ 1, 2, 3 ]
[ 4, 5, 6 ]
[ 7, 8, 9 ]
15
*/
let readline = require("readline-sync");
let array = [];
let num = 1;
let sum = 0;
console.log("请输入矩形的宽：");
let input = parseInt(readline.question(""));
for (let i = 0; i < input; i++) { 
    array[i] = [];
    for (let k = 0; k < input; k++) { 
        array[i][k] = num++;
    }
}
for (let i = 0; i < input; i++) { 
    console.log(array[i]);
    sum = sum + array[i][i];
}

console.log(sum);