JavaScript 中具有 0 和 1 的连续子数组

Z时代
2024-01-10
分类：综合

问题：

我们需要编写一个 JavaScript 函数，它接受一个二进制数组 arr（一个只包含 0 或 1 的数组）。我们的函数应该从由相同数量的 1 和 0 组成的数组中返回连续子数组的长度。

例如，如果函数的输入是 -

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

那么输出应该是 -

const output = 6;

输出说明

数组的前6个元素是1, 0, 0, 1, 0, 1（三个1和三个0）

示例

此代码将是 -

const arr = [1, 0, 0, 1, 0, 1, 0, 0];
const findMaxLength = (arr = []) => {
   const { length } = arr;
   if (length < 2){
      return 0
   };
   const map = new Map();
   map.set(0, -1);
   let sum = 0;
   let max = 0;
   for (var i = 0; i < length; i++) {  
      sum += arr[i] === 0 ? -1 : 1;
      if (map.has(sum)) {
         max = Math.max(max, i - map.get(sum));
      } else {
         map.set(sum, i);
      };
   };
   return max;
};
console.log(findMaxLength(arr));