计数集在整数数组中具有相同的差异

Question

有一个数组：[1,1,1,1,2,3,4]，计算数组中所有相同差异的集合 ※数组中每个集合的差值可以不一样

Way to count one set have same difference:
-1,0,1  -> count = 1 with diff = 1
1,1,1   -> count = 1 with diff = 0
1,2,3,4 -> count = 3 with diff = 1 ([1,2,3],[1,2,3,4],[2,3,4])
0,1     -> not count
0,1,3   -> not count
0,1,1,2 -> not count , althought it has 0,1 and 1,2 with diff = 1 but it is not continous

As the above array  [1,1,1,1,2,3,4] count = 6
There are 6 sets has the same difference value:
 [1,1,1],[1,1,1,1],[1,1,1] with difference = 0
 [1,2,3],[1,2,3,4],[2,3,4] with difference = 1

这是我的蛮力方法 O(n2) 我可以使用带记忆的 DP 来更有效地改进它吗？

 function run(pArray){
      var count = 0;
      var length = pArray.length;
      for(var i = 0;i<length - 2;i++){          
        var nextIdx = i + 1;
        var diff = pArray[nextIdx]-pArray[i];          
        while(true){
          nextIdx ++;
        
          if(nextIdx > length - 1){
            break;
          }    
          if(pArray[nextIdx] - pArray[nextIdx - 1] !== diff){
            break;
          }else{
            count ++;
          }
        }
    }
    
    console.log(count);
    }
    
    run([1,1,1,1,2,3,4]);

Answer 1

根据您的最新评论，您似乎确实关心数组中元素的顺序。因此不需要对其进行排序，在这种情况下，集合的数量可以评估为零，如 [1,2,1,2,1] 的情况。

在这种情况下，它可以在O(n) 中完成。

想法是：

如果你发现一个匹配的差异增加一个计数变量，更新 结果为结果 = 结果 + 计数。

举个例子[1,1,1,1]。您无需考虑索引 0,1。

取索引 2 看看是否 arr[2] - arr[1] == arr[1] - arr[0] 然后递增 从 0 到 1 计数，将其添加到结果中。

取索引 3 看看是否 arr[3] - arr[2] == arr[2] - arr[1] ，是的 相等，然后将 count 增加到 2 ，将其添加到结果中。现在结果 = 1 + 2 = 3。

只要你得到的元素满足 标准 arr[i] - arr[i-1] == arr[i-1] - arr[i-2] 你不断添加 下一个更高的价值。所以结果 = 1 + 2 + 3 +....i-2 以此类推 给定索引 i

如果您没有找到匹配的差异，请将计数变量重置为零 并重新开始。

注意：我对 Javascript 不太熟悉，但它看起来像：

<div>
    <script>
      function countSets(arr) {
        if (arr === undefined || arr.length < 3) return 0;
        let count = 0;
        let res = 0;
        for (let i = 2; i < arr.length; ++i) {
          if (arr[i] - arr[i - 1] === arr[i - 1] - arr[i - 2]) {
            count++;
          } else {
            count = 0;
          }
          res += count;
        }
        return res;
      }
      console.log("[1, 1, 1] -> " + countSets([1, 1, 1]))
      console.log("[1, 1, 1, 2, 2, 2] -> " + countSets([1, 1, 1, 2, 2, 2]))
      console.log("[1, 1, 1, 2, 3, 4] -> " + countSets([1, 1, 1, 2, 3, 4]))
      console.log("[1, 1, 2, 3, 4, 4, 5, 5] -> " + countSets([1, 1, 2, 3, 4, 4, 5, 5]))
      console.log("[1, 1, 1, 2, 3, 4, 5, 6] -> " + countSets([1, 1, 1, 2, 3, 4, 5, 6]))
      console.log("[1, 2, 1, 2, 1] -> " + countSets([1, 2, 1, 2, 1]))
    </script>
</div>