Get Array B from Array A Counting Backwards in JavaScript

Consider the following binary array (Array A) −

const arr = [1,0,1,1,1,1,0,1,1];

When this array is passed through the function, say sumRight(), it produces the following output array (Array B) −

const output = [1,0,4,3,2,1,0,2,1];

Understanding the function

Elements in array arr can be either 0 or 1. The function counts backward from the last element of array arr, if there are consecutive 1's in the array arr then the corresponding element in the output array will be 1 but for the 2nd consecutive 1 in array arr, it will be 2. For the 3rd one in input array the element in the output array will be 3, but for 0 in the array arr it will be 0 in the output array as well.

So let’s write the code for this function using the Array.prototype.reduceRight() method, which does the same work as the normal reduce method, it just starts from the right instead of left −

Example

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));

Output

The output in the console will be −

[
   1, 0, 4, 3, 2,
   1, 0, 2, 1
]