请看以下二进制数组(数组A)-
const arr = [1,0,1,1,1,1,0,1,1];
当此数组通过函数传递时,例如sumRight(),它将产生以下输出数组(数组B)-
const output = [1,0,4,3,2,1,0,2,1];
数组arr中的元素可以为0或1。该函数从数组arr的最后一个元素开始向后计数,如果数组arr中有连续的1,则输出数组中的对应元素将为1,但对于第二个连续的1数组arr为2。对于输入数组中的第3个元素,输出数组中的元素为3,但对于数组arr中的0,其输出数组中的元素也为0。
因此,让我们使用Array.prototype.reduceRight()方法编写此函数的代码,该方法的功能与普通的reduce方法相同,只是从右开始而不是从左开始-
const arr = [1,0,1,1,1,1,0,1,1];
const sumRight = arr => {
return arr.reduceRight((acc, val) => {
const { prev, res } = acc;
if(val === 0){
return {
prev: 0,
res: res.concat(0)
};
};
return {
res: res.concat(val+prev),
prev: prev+1
};
}, {
prev: 0,
res: []
}).res.reverse();
};
console.log(sumRight(arr));输出结果
控制台中的输出将为-
[ 1, 0, 4, 3, 2, 1, 0, 2, 1 ]