如何求二维矩阵 ( m X n) 中每行和每列的总和。
[
[1, 2, 3],
[3, 2, 1]
]
我知道在一维数组中,我们可以这样做:
var sum = [5, 6, 3].reduce(add, 0);
function add(a, b) {
return a + b;
}
console.log(sum);
使用 ECMAScript6,您可以执行以下操作:
var matrix = [
[ 1, 2, 3 ],
[ 7, 2, 6 ]
];
// sums of rows
var rowSum = matrix.map(r => r.reduce((a, b) => a + b));
// sums of columns
var colSum = matrix.reduce((a, b) => a.map((x, i) => x + b[i]));
console.log(rowSum);
console.log(colSum);
行的总和很容易说明。让我们考虑一个更容易格式化的 3x3 矩阵:
[ 1 2 3 ] -- reduce --> 6 \
[ 7 2 6 ] -- reduce --> 15 } -- map --> [ 6, 15, 7 ]
[ 4 1 2 ] -- reduce --> 7 /
列的总和就不那么简单了。我们通过一次“映射”两行来“减少”:
[ 1 2 3 ] [ 8 4 9 ]
+ [ 7 2 6 ] -- reduce --> + [ 4 1 2 ]
--------- ---------
map = [ 8 4 9 ] map = [ 12 5 11 ]
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)