JAVA程序设计：三数之和的多种可能（LeetCode：923）

给定一个整数数组 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

思路：由于target的值很小，因此我们考虑枚举前两个数组，然后通过map查找第三个数字的数量即可，为了避免重复计数，我们需要保证第二个数不小于第一个数，第三个数不小于第二个数。

class Solution {
	
	private int mod=1000000007;
	
    public int threeSumMulti(int[] A, int target) {
    	
    	long ans=0;
    	Map<Integer,Long> map=new HashMap<>(); 
    	
    	for(int i=0;i<A.length;i++)
    		map.put(A[i], map.getOrDefault(A[i], 0L)+1);
    	
    	for(int i=0;i<=target;i++)
    		for(int j=i;j<=target;j++) {
    			if(i+j>target || target-i-j<j) break;

    			long val1=map.getOrDefault(i, 0L);
    			long val2=map.getOrDefault(j, 0L);
    			long val3=map.getOrDefault(target-i-j, 0L);
    			
    			if(val1==0 || val2==0 || val3==0) continue;
    			
    			if(target-i-j==i) 
    				ans=(ans+val1*(val1-1)*(val1-2)/6%mod)%mod;
    			else if(i==j)
    				ans=(ans+val1*(val1-1)/2*val3%mod)%mod;
    			else if(target-i-j==j)
    				ans=(ans+val2*(val2-1)/2*val1%mod)%mod;
    			else
    				ans=(ans+val1*val2*val3%mod)%mod;
    		}
    	
    	return (int)ans;
    }
}