AGC 001E.BBQ Hard(组合 DP)

时间:2023-03-09 09:56:03
AGC 001E.BBQ Hard(组合 DP)

题目链接

\(Description\)

给定长为\(n\)的两个数组\(a,b\),求$$\sum_{i=1}n\sum_{j=i+1}n\binom{a_i+a_j+b_i+b_j}{a_i+a_j}\ mod\ 10^9+7$$

\(n\leq2\times10^5,a_i,b_i\leq2000\)。

\(Solution\)

考虑\(\binom{n+m}{n}\)的组合意义:在网格图上从\((0,0)\)走到\((n,m)\)的方案数(NE lattice paths)。也等于从\((x,y)\)走到\((x+n,y+m)\)的方案数。

考虑\(\binom{a_i+a_j+b_i+b_j}{a_i+a_j}\)的组合意义:从\((0,0)\)走到\((a_i+a_j+b_i+b_j)\)的方案数,从\((-a_i,-b_i)\)走到\((a_j,b_j)\)的方案数。

考虑计算从任意\((-a_i,-b_i)\)走到任意\((a_j,b_j)\)的方案数。从某个点走到另一个点的方案数可以递推得到。从某些点走到某个点的方案数同样可以递推。

算完后减去每个点自己到自己的方案数,再把答案除以\(2\)就行了。

复杂度\(O((2\max(a_i,b_i))^2)\)。

//127ms	64384KB
#include <cstdio>
#include <cctype>
#include <algorithm>
#define gc() getchar()
#define mod 1000000007
#define Mod(x) x>=mod&&(x-=mod)
typedef long long LL;
const int N=2e5+5,M=4005,D=2001; int A[N],B[N],f[M][M],fac[N],ifac[N]; inline int read()
{
int now=0;register char c=gc();
for(;!isdigit(c);c=gc());
for(;isdigit(c);now=now*10+c-'0',c=gc());
return now;
}
inline int FP(int x,int k)
{
int t=1;
for(; k; k>>=1,x=1ll*x*x%mod)
if(k&1) t=1ll*t*x%mod;
return t;
}
#define C(n,m) 1ll*fac[n]*ifac[m]%mod*ifac[(n)-(m)]%mod int main()
{
int n=read(),mx=0;
for(int i=1,x,y; i<=n; ++i)
{
A[i]=x=read(), B[i]=y=read(), ++f[D-x][D-y];//(-x+D,-y+D)
mx=std::max(mx,std::max(x,y));
}
int lim=D+mx;
for(int i=D-mx; i<=lim; ++i)
for(int j=D-mx; j<=lim; ++j)
f[i][j]+=f[i-1][j]+f[i][j-1], Mod(f[i][j]);//a little bigger LL ans=0;
for(int i=1; i<=n; ++i) ans+=f[D+A[i]][D+B[i]];
ans%=mod; lim=mx<<2, fac[0]=fac[1]=1;
for(int i=2; i<=lim; ++i) fac[i]=1ll*fac[i-1]*i%mod;
ifac[lim]=FP(fac[lim],mod-2);
for(int i=lim-1; ~i; --i) ifac[i]=1ll*ifac[i+1]*(i+1)%mod;
for(int i=1; i<=n; ++i) ans-=C(A[i]*2+B[i]*2,A[i]*2); ans%=mod, ans*=FP(2,mod-2);
printf("%lld\n",(ans%mod+mod)%mod); return 0;
}