矩阵对角线元素的和

Desc

给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。

请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。

Code

1
2
3
4
5
6
7
8
9
10
11
12

var diagonalSum = function(mat) {
if(mat.length === 1) {
return mat[0][0];
}
return mat.reduce((count, v, i) => {
if(i === (v.length - 1 - i)) {
return count + v[i];
}
return count + (v[i] + v[v.length - 1 - i]);
}, 0);
};