[agc001e]BBQ hard

最新推荐文章于 2022-03-13 16:30:19 发布

WerKeyTom_FTD

最新推荐文章于 2022-03-13 16:30:19 发布

阅读量678

点赞数

分类专栏：排列组合

本文链接：https://blog.csdn.net/WerKeyTom_FTD/article/details/77996280

版权

排列组合专栏收录该内容

51 篇文章 0 订阅

订阅专栏

题目大意

给你a[]和b[]，权值范围在2e3内。
求 $\sum_{i=1}^{n-1}\sum_{j=i+1}^nC_{a_i+a_j+b_i+b_j}^{a_i+a_j}$

做法

你考虑 $C_{a+b}^b$ 的组合意义其实是从(0,0)走到(a,b)每次只能向上和向右有多少走法。
因此这个魔幻的式子其实相当于(0,0)走到两个向量和的位置，但是并不能暴力枚举这两个向量。
考虑向量可以平移，即在(-a,-b)走到(a’,b’)的方案数也是 $C_{a+b+a'+b'}^{a+a'}$
所以就可以做了

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int maxn=200000+10,maxv=5000+10,mo=1000000007,inv2=(mo+1)/2;
int c[maxv][maxv],a[maxn],b[maxn],fac[maxv*2],inv[maxv*2];
int i,j,k,l,t,n,m,ans,mxa,mxb;
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
int qsm(int x,int y){
    if (!y) return 1;
    int t=qsm(x,y/2);
    t=(ll)t*t%mo;
    if (y%2) t=(ll)t*x%mo;
    return t;
}
int C(int n,int m){
    if (m<0||n<m) return 0;
    return (ll)fac[n]*inv[m]%mo*inv[n-m]%mo;
}
int main(){
    freopen("a.in","r",stdin);freopen("a.out","w",stdout);
    n=read();
    fo(i,1,n){
        a[i]=read();b[i]=read();
        mxa=max(mxa,a[i]);mxb=max(mxb,b[i]);
    }
    fac[0]=1;
    fo(i,1,mxa*2+mxb*2) fac[i]=(ll)fac[i-1]*i%mo;
    inv[mxa*2+mxb*2]=qsm(fac[mxa*2+mxb*2],mo-2);
    fd(i,mxa*2+mxb*2-1,0) inv[i]=(ll)inv[i+1]*(i+1)%mo;
    fo(i,1,n) c[mxa-a[i]][mxb-b[i]]++;
    fo(i,0,mxa*2)
        fo(j,0,mxb*2){
            if (j) (c[i][j]+=c[i][j-1])%=mo;
            if (i) (c[i][j]+=c[i-1][j])%=mo;
        }
    fo(i,1,n) (ans+=c[mxa+a[i]][mxb+b[i]])%=mo;
    fo(i,1,n) (ans-=C(a[i]*2+b[i]*2,a[i]*2))%=mo;
    ans=(ll)ans*inv2%mo;
    (ans+=mo)%=mo;
    printf("%d\n",ans);
}