leetcode923. 三数之和的多种可能

给定一个整数数组 arr ，以及一个整数 target 作为目标值，返回满足 i < j < k 且 arr[i] + arr[j] + arr[k] == target 的元组 i, j, k 的数量。

由于结果会非常大，请返回 109 + 7 的模。

示例 1：

输入：arr = [1,1,2,2,3,3,4,4,5,5], target = 8
输出：20
解释：
按值枚举(arr[i], arr[j], arr[k])：
(1, 2, 5) 出现 8 次；
(1, 3, 4) 出现 8 次；
(2, 2, 4) 出现 2 次；
(2, 3, 3) 出现 2 次。

示例 2：

输入：arr = [1,1,2,2,2,2], target = 5
输出：12
解释：
arr[i] = 1, arr[j] = arr[k] = 2 出现 12 次：
我们从 [1,1] 中选择一个 1，有 2 种情况，
从 [2,2,2,2] 中选出两个 2，有 6 种情况。

提示：

• 3 <= arr.length <= 3000
• 0 <= arr[i] <= 100
• 0 <= target <= 300

  /**
   * @param {number[]} arr
   * @param {number} target
   * @return {number}
   */
   let mod_num =(10**9) + 7;
  var threeSumMulti = function(arr, target) {
      arr.sort((a,b) => a - b);
      let n = arr.length;
      let ans = 0;
      for(let i = 0,I = n - 2;i < I;i++){
          ans += twoSumMulti(arr,i + 1,n - 1,target - arr[i]);
          ans %= mod_num;
      }
      return ans;
  };
  var twoSumMulti = function(arr,l,r, target) {
      let ans = 0;
      while(l < r){
          if(arr[l] + arr[r] < target) l += 1;
          else if(arr[l] + arr[r] > target) r-= 1;
          else{
              if(arr[l] == arr[r]){
                  let n = r - l + 1;
                  ans += n * (n - 1) / 2;
                  ans %= mod_num;
                  break;
              }
              let lcnt = 1,rcnt = 1;
              while(arr[l + 1] == arr[l]){
                  lcnt += 1;
                  l += 1;
              }
              while(arr[r- 1] == arr[r]){
                  rcnt += 1;
                  r -= 1;
              }
              ans += lcnt * rcnt;
              ans %= mod_num;
              l += 1,r -= 1; 
          }
      }
      return ans;
  };