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;
}

猜你喜欢

转载自www.cnblogs.com/SovietPower/p/9750690.html