题意:
有n个包,一个包里面有一根竹签,上面有编号i,还有Ai个A物品,Bi个B物品。现在选择两个包,用两个竹签将A物品和B物品串起来。两种方法是不一样的,当且仅当选择的竹签的编号不同(忽略顺序)或者A,B物品的摆放顺序不同(可重复排列)。
下面是N=3的情况:
数据范围:
2≦N≦200,000
1≦Ai≦2000,1≦Bi≦2000
思路:
考试的时候只会骗…首先,很容易想到一个O(n^2)的算法,也就是枚举两个包,然后算这两个包的可重复排列,就是一道大版题了…这样子可以骗60分(够了够了 )。
其实我们通常在使用组合数的时候,都忽略了组合数的基本定义。这道题的答案可以是:
∑
i
=
1
N
∑
j
=
i
+
1
N
C
A
i
+
B
i
+
A
j
+
B
j
A
i
+
A
j
\sum_{i=1}^{N}\sum_{j=i+1}^{N}C_{A_i+B_i+A_j+B_j}^{A_i+A_j}
i=1∑Nj=i+1∑NCAi+Bi+Aj+BjAi+Aj
然后可以转化为一个经典组合问题:从(-Ai,-Bi)到(Aj,Bj),只能往右或往上走的方案数。
这样就将每一个(-Ai,-Bi)的dp值初始化为1,然后从左下向右上递推就可以了。
还需要注意的是,这样算出来还需要去重。
A
n
s
=
∑
i
=
1
N
∑
j
=
i
+
1
N
C
A
i
+
B
i
+
A
j
+
B
j
A
i
+
A
j
Ans=\sum_{i=1}^{N}\sum_{j=i+1}^{N}C_{A_i+B_i+A_j+B_j}^{A_i+A_j}
Ans=∑i=1N∑j=i+1NCAi+Bi+Aj+BjAi+Aj
=
(
∑
i
=
1
N
∑
j
=
1
N
C
A
i
+
B
i
+
A
j
+
B
j
A
i
+
A
j
−
∑
i
=
1
N
C
2
A
i
+
2
B
i
A
i
+
B
i
)
2
=\dfrac{(\sum_{i=1}^{N}\sum_{j=1}^{N}C_{A_i+B_i+A_j+B_j}^{A_i+A_j}-\sum_{i=1}^{N}C_{2A_i+2B_i}^{A_i+B_i})}{2}
=2(∑i=1N∑j=1NCAi+Bi+Aj+BjAi+Aj−∑i=1NC2Ai+2BiAi+Bi)
然后就可以计算答案了。
其实我觉得这道题的正解比较难想,感觉是出题人想到可以转换这个模型之后,才硬搞出来的这样一道题目。当然这只是个人的看法,反正我在看到正解的做法的时候是很震惊的…
#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXN 200000
#define MAXV 4000
#define MO 1000000007
using namespace std;
int n,A[MAXN+5],B[MAXN+5];
int dp[MAXV+5][MAXV+5];
int fact[MAXV*2+5],inv[MAXV*2+5];
int PowMod(int a,int b)
{
int ret=1;
while(b)
{
if(b&1)
ret=1LL*ret*a%MO;
a=1LL*a*a%MO;
b>>=1;
}
return ret;
}
void prepare()
{
fact[0]=1;
for(int i=1;i<=MAXV*2;i++)
fact[i]=1LL*fact[i-1]*i%MO;
inv[MAXV*2]=PowMod(fact[MAXV*2],MO-2);
for(int i=MAXV*2-1;i>=0;i--)
inv[i]=1LL*inv[i+1]*(1LL*i+1LL)%MO;
}
int C(int a,int b)
{
return 1LL*fact[a]*inv[b]%MO*inv[a-b]%MO;
}
int main()
{
// freopen("bbq.in","r",stdin);
// freopen("bbq.out","w",stdout);
prepare();
scanf("%d",&n);
int maxval=-1;
for(int i=1;i<=n;i++)
{
scanf("%d %d",&A[i],&B[i]);
maxval=max(maxval,A[i]);
maxval=max(maxval,B[i]);
}
maxval++;
for(int i=1;i<=n;i++)
dp[maxval-A[i]][maxval-B[i]]++;
for(int i=-maxval+1;i<=maxval;i++)
for(int j=-maxval+1;j<=maxval;j++)
dp[i+maxval][j+maxval]=(1LL*dp[i+maxval][j+maxval]+1LL*dp[i-1+maxval][j+maxval]+1LL*dp[i+maxval][j-1+maxval])%MO;
int ans=0;
for(int i=1;i<=n;i++)
{
ans=(1LL*ans+1LL*dp[A[i]+maxval][B[i]+maxval])%MO;
ans=((1LL*ans-1LL*C(A[i]+B[i]+A[i]+B[i],A[i]+A[i]))%MO+MO)%MO;
}
ans=1LL*ans*PowMod(2,MO-2)%MO;
printf("%d\n",ans);
return 0;
}