AT1983-[AGC001E]BBQ Hard【dp,组合数学】

正题

题目链接:https://www.luogu.com.cn/problem/AT1983

---

题目大意

给出(n)个数对((a_i,b_i))

求

[sum_{i=1}^nsum_{j=i+1}^ninom{a_i+b_i+a_j+b_j}{a_i+a_j} ]

(1leq nleq 2 imes 10^5,1leq a_i,b_ileq 2000)

---

解题思路

啊遇到这种题目直接上组合意义 (color{white}{组合意义天地灭})

然后发现(a_{i},b_i)很小。上面那个组合数可以变成横着走(a_i+a_j)步，竖着走(b_i+b_j)步的方案。

之后理解为从((-a_i,-b_i))走到((a_j,b_j))就可以分离(i,j)了。

因为很小，直接多起点走一次求和就好了，要减去重复的部分。

时间复杂度(O(n+max{a_i} imes max{b_i}))

---

code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=2e5+10,M=4500,P=1e9+7;
ll n,a[N],b[N],f[M][M],fac[N],inv[N],ans;
ll C(ll n,ll m)
{return fac[n]*inv[m]%P*inv[n-m]%P;}
signed main()
{
	scanf("%lld",&n);inv[1]=1;
	for(ll i=2;i<N;i++)inv[i]=P-inv[P%i]*(P/i)%P;
	inv[0]=fac[0]=1;
	for(ll i=1;i<N;i++)
		fac[i]=fac[i-1]*i%P,inv[i]=inv[i-1]*inv[i]%P;
	for(ll i=1;i<=n;i++){
		scanf("%lld%lld",&a[i],&b[i]);
		f[2001-a[i]][2001-b[i]]++;
	}
	for(ll i=1;i<=4002;i++)
		for(ll j=1;j<=4002;j++)
			(f[i][j]+=f[i-1][j]+f[i][j-1])%=P;
	for(ll i=1;i<=n;i++){
		(ans+=f[2001+a[i]][2001+b[i]]%P)%=P;
		(ans-=C(2*a[i]+2*b[i],2*a[i]))%=P;
	}
	printf("%lld
",(ans+P)*inv[2]%P);
	return 0;
}