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

示例 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

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;
                    else
                        count += (n1+n2)*(n1+n2-1)/2;
    				count %= mod;
    			}
    			else if(sum < target)
    				j++;
    			else
    				k--;
    		}
    	}
    	return count;
    }
};