LeetCode 923. 三数之和的多种可能（双指针）

1. 题目

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

由于结果会非常大，请返回结果除以 10^9 + 7 的余数。

示例 1：
输入：A = [1,1,2,2,3,3,4,4,5,5], target = 8
输出：20
解释：
按值枚举（A[i]，A[j]，A[k]）：
(1, 2, 5) 出现 8 次；
(1, 3, 4) 出现 8 次；
(2, 2, 4) 出现 2 次；
(2, 3, 3) 出现 2 次。示例 2：
输入：A = [1,1,2,2,2,2], target = 5
输出：12
解释：
A[i] = 1，A[j] = A[k] = 2 出现 12 次：
我们从 [1,1] 中选择一个 1，有 2 种情况，
从 [2,2,2,2] 中选出两个 2，有 6 种情况。提示：
3 <= A.length <= 3000
0 <= A[i] <= 100
0 <= target <= 300

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum-with-multiplicity
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

先排序
固定第一个数 i，然后 j=i+1, k = n-1
三数和不为target，则相应的左右端点，移动
三数和为target，左右端点旁边可能有相等，都找出来，左右相乘（左右数值不相等）
三数和为target，左右相等的话，左右的总个数取2个的组合数

class Solution {
public:int threeSumMulti(vector<int>& A, int target) {int i, j, k, t, n = A.size(), mod = 1e9+7, count = 0, sum, l, r, n1, n2;sort(A.begin(), A.end());for(i = 0; i < n-2; ++i){j = i+1, k = n-1;while(j < k){sum = A[i]+A[j]+A[k];if(sum==target){l = A[j], r = A[k];n1 = n2 = 0;while(j < k && A[j]==l)n1++,j++;while(j <= k && A[k]==r)//等号不能少，上面j++后可能到k了n2++,k--;if(l != r)count += n1*n2;elsecount += (n1+n2)*(n1+n2-1)/2;count %= mod;}else if(sum < target)j++;elsek--;}}return count;}
};