给定一个整数数组 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
题目难度很小,主要是考察了个简单的组合数学应用
题目范围已经很明确了,
A
[
i
]
A[i]
A[i]才0到100,用一个数组记录一下每个数出现的次数。然后分三种情况讨论,最后求和就可以了
1、
三个数一样,并且当
t
a
r
g
e
t
=
3
∗
n
(
n
∈
N
∗
)
target=3*n (n∈N^*)
target=3∗n(n∈N∗)时:
r
e
s
u
l
t
1
=
C
n
3
,
n
=
t
a
r
g
e
t
/
3
result_1=C_{n}^{3},n=target/3
result1=Cn3,n=target/3
2、
两个数一样,
r
e
s
u
l
t
2
=
C
p
2
∗
C
q
1
,
p
+
q
=
3
,
p
>
0
,
q
>
0
result_2=C_{p}^{2}*C_{q}^{1},p+q=3,p>0,q>0
result2=Cp2∗Cq1,p+q=3,p>0,q>0
3、
三个数互不相同,
r
e
s
u
l
t
3
=
C
p
1
∗
C
q
1
∗
C
r
1
,
p
=
q
=
r
=
1
result_3=C_{p}^{1}*C_{q}^{1}*C_{r}^{1},p=q=r=1
result3=Cp1∗Cq1∗Cr1,p=q=r=1
然后答案为:
r
e
s
u
l
t
=
r
e
s
u
l
t
1
+
r
e
s
u
l
t
2
+
r
e
s
u
l
t
3
result=result_1+result_2+result_3
result=result1+result2+result3
为了计算方便,提前把
C
3000
3
C_{3000}^{3}
C30003以内的组合数都预处理出来了
class Solution{
private:
long long C[3005][4];
const long long MOD=1000000007;
public:
Solution()
{
memset(C,0,sizeof(C));
for(int i=0;i<=3000;++i)
{
for(int j=0;j<=3&&j<=i;++j)
{
if(j==0)
C[i][j]=1;
else
C[i][j]=(C[i-1][j-1]%MOD+C[i-1][j]%MOD)%MOD;
}
}
}
int threeSumMulti(vector<int>& A, int target)
{
long long result=0;
int hc[305]={0};
for(int i=0;i<A.size();++i)
{
hc[A[i]]+=1;
}
if(target%3==0)
{
int x=target/3;
if(hc[x]>=3)
result=(result+C[hc[x]][3])%MOD;
}
for(int i=0;i<=100;++i)
{
if(hc[i]>=2 && 2*i<=target)
{
int x=target-2*i;
if(x==i || hc[x]==0)
continue;
result=(result+(C[hc[i]][2]%MOD*C[hc[x]][1]%MOD)%MOD)%MOD;
}
}
for(int i=0;i<=100;++i)
{
if(hc[i]>=1)
{
for(int j=i+1;j<=100;++j)
{
if(hc[j]>=1)
{
int x=target-i-j;
if(x<=j)
break;
if(x==i || x==j || hc[x]==0)
continue;
result=(result+(((C[hc[i]][1]*C[hc[j]][1])%MOD)*C[hc[x]][1])%MOD)%MOD;
}
}
}
}
return result%MOD;
}
};