给定一个整数数组 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 种情况。
思路:三指针法
特殊之处在于相同数字也算不同的个数,所以每次需要判断有没有相同组成的数组之和也等于target
情况一:nums[start]与nums[end]不一样,例如示例一,1,2,5时,i=0时,通过判断,2出现2次,5出现2次,所以2*2=4次
情况二:nums[start]与nums[end]一样,例如示例二,1,2,2时,i=0时,后面4个2,选择两个即可,所以是4*3/2=6种
提示:
3 <= A.length <= 3000
0 <= A[i] <= 100
0 <= target <= 300
提交的代码:
class Solution {
public int threeSumMulti(int[] nums, int target) {
Arrays.sort(nums);
int i,start=1,end=nums.length-1,sum=0,a,b,c,t;
for(i=0;i<nums.length-2;i++)
{
a=0;
b=0;
c=0;
start=i+1;
end=nums.length-1;
while(start<end)
{
a=0;
b=0;
c=0;
if(nums[i]+nums[start]+nums[end]>target)
{
end--;
}
else if(nums[i]+nums[start]+nums[end]<target)
{
start++;
}
else
{
while(nums[start]==nums[start+1]&&start+1!=end)
{
if(nums[start]==nums[end])
{
c++;
}
start++;
a++;
}
while(nums[end]==nums[end-1]&&end-1!=start)
{
end--;
b++;
}
if(a==0&&b==0)
{
sum++;
}
else{
if(nums[start]!=nums[end])
{
sum= sum+(a+1)*(b+1);
}
else
{
c+=2;
t = c*(c-1)/2;
sum+=t;
}
}
sum=sum%1000000007;
start++;
end--;
}
}
}
return sum;
}
}