题意:给定$n$个烧烤套装,每个套装里有$1$个竹签,$A_i$片牛肉和$B_i$片青椒,竹签两两不同,食材都是相同的,现在要用两个套装串一串(两个竹签串上这两个套装的所有食材),问能串出多少种不同的串
答案是$\sum\limits_{1\leq i\lt j\leq n}\binom{a_i+b_i+a_j+b_j}{a_i+a_j}$,直接求会T
因为我们求的组合数都是$\binom{a+b}a$的形式,不妨想一下它的组合意义——从$(0,0)$走到$(a,b)$的方案数
那么$\binom{a_i+b_i+a_j+b_j}{a_i+a_j}$就是从$(-a_i,-b_i)$走到$(a_j,b_j)$的方案数,我们可以令所有的$(-a_i,-b_i)$为起点,DP求得走到第一象限的点的方案数,那么答案就是所有走到$(a_i,b_i)$的方案数减去$\binom{2a_i+2b_i}{2a_i}$之和再$\div2$(自己到自己不应该被计数,$a\rightarrow b$和$b\rightarrow a$应该是同样的方案)
这个题的思想挺棒的!
#include<stdio.h>
typedef long long ll;
const int mod=1000000007,i2=(mod+1)/2;
int mul(int a,int b){return(ll)a*b%mod;}
void inc(int&a,int b){(a+=b)%=mod;}
int pow(int a,int b){
int s=1;
while(b){
if(b&1)s=mul(s,a);
a=mul(a,a);
b>>=1;
}
return s;
}
int fac[8010],rfac[8010];
int C(int n,int k){return mul(fac[n],mul(rfac[k],rfac[n-k]));}
int a[200010],b[200010],f[4010][4010];
int main(){
int n,i,j,s;
fac[0]=1;
for(i=1;i<=8000;i++)fac[i]=mul(fac[i-1],i);
rfac[8000]=pow(fac[8000],mod-2);
for(i=8000;i>0;i--)rfac[i-1]=mul(rfac[i],i);
scanf("%d",&n);
for(i=1;i<=n;i++){
scanf("%d%d",a+i,b+i);
f[2000-a[i]][2000-b[i]]++;
}
for(i=0;i<=4000;i++){
for(j=0;j<=4000;j++){
inc(f[i][j+1],f[i][j]);
inc(f[i+1][j],f[i][j]);
}
}
s=0;
for(i=1;i<=n;i++)inc(s,f[2000+a[i]][2000+b[i]]-C((a[i]+b[i])*2,a[i]*2));
inc(s,mod);
printf("%d",mul(s,i2));
}