JavaScript 中包含 n 个不同元素的数组部分


问题

我们需要编写一个 JavaScript 函数,此函数以一个文字数组 arr 作为第一个参数。第二个参数将是一个数字 num。我们的函数应统计并返回数组中包含恰好 num 个不同元素的子数组数。

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

const arr = [12, 15, 12, 15, 18];
const num = 2;

那么输出应该为 -

const output = 7;

输出说明

精确由 2 个不同元素形成的子数组 -

[12,15], [15,12], [12,15], [15,18], [12,15,12], [15,12,15], [12,15,12,15]

示例

代码如下 -

 在线演示

const arr = [12, 15, 12, 15, 18];
const num = 2;
const distinctSubarrays = (arr = [], num = 1) => {
   const findDistinct = (count) => {
      const map = {};
      let ptr = 0;
      let distinct = 0;
      let res = 0;
      for(let right = 0; right < arr.length; right++){
         const num = arr[right];
         map[num] = (map[num] || 0) + 1;
         if(map[num] === 1){
            distinct += 1;
         };
         while(distinct > count){
            map[arr[ptr]] -= 1;
            if(map[arr[ptr]] === 0){
               distinct -= 1;
            };
            ptr += 1;
         };
         res += right - ptr + 1;
      };
      return res;
   };
   return findDistinct(num) - findDistinct(num - 1)
};
console.log(distinctSubarrays(arr, num));

输出

控制台中输出的内容为 -

7

更新于: 09-4-2021

119 次浏览

启动你的事业

完成课程获得认证

开始
广告