#dp,模型转换,排列组合#AT1983 [AGC001E] BBQ Hard

题目

有两个长度为(n)的序列(a,b),需要求

[sum_{i=1}^nsum_{j=i+1}^nC(a_i+b_i+a_j+b_j,a_i+a_j) ]

其中(nleq 200000,a,bleq 2000)


分析

突破口在(a,b)的小范围,观察后面这一坨东西
也就是一个长宽分别为(a_i+a_j,b_i+b_j)的矩阵
从左下角走到右上角只能通过向上或者向右的方法,这东西可以用(dp)解决
那我如果可以把所有((a_i,b_i))扔到(dp)里解决自然就搞定了,考虑将坐标平移
那么也就是求((-a_i,-b_i))((a_j,b_j))的方案数,
那这样互不干扰给每一个起点加1,直接大力(dp)
注意先要减去((-a_i,-b_i))((a_i,b_i))的方案然后再除以2(按照题意)


代码

#include<cstdio>
#include<cctype>
#define rr register
using namespace std;
const int M=4003,N=200011,mod=1000000007;
int dp[M][M],n,fac[M<<1],inv[M<<1],a[N],b[N],ans;
inline signed iut(){
    rr int ans=0; rr char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) ans=ans*10+c-48,c=getchar();
    return ans;
}
inline signed mo(int x,int y){return x+y>=mod?x+y-mod:x+y;}
inline signed c(int n,int m){return 1ll*fac[n]*inv[m]%mod*inv[n-m]%mod;}
signed main(){
    n=iut(),fac[0]=fac[1]=inv[0]=inv[1]=1;
    rr int t=M>>1,T=t<<2;
    for (rr int i=2;i<=T;++i) inv[i]=1ll*inv[mod%i]*(mod-mod/i)%mod;
    for (rr int i=2;i<=T;++i) fac[i]=1ll*fac[i-1]*i%mod,inv[i]=1ll*inv[i-1]*inv[i]%mod;
    for (rr int i=1;i<=n;++i){
        a[i]=iut(),b[i]=iut();
        ++dp[t-a[i]][t-b[i]];
    }
    for (rr int i=1;i<M;++i)
    for (rr int j=1;j<M;++j){
        rr int w=mo(dp[i-1][j],dp[i][j-1]);
        dp[i][j]=mo(dp[i][j],w);
    }
    for (rr int i=1;i<=n;++i){
        ans=mo(ans,dp[t+a[i]][t+b[i]]);
        ans=mo(ans,mod-c(a[i]*2+b[i]*2,a[i]*2));
    }
    return !printf("%d
",1ll*ans*inv[2]%mod);
}
原文地址:https://www.cnblogs.com/Spare-No-Effort/p/13763269.html