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

1|0正题

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

---

1|1题目大意

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

求

$$\sum_{i=1}^n\sum_{j=i+1}^n\binom{a_i+b_i+a_j+b_j}{a_i+a_j}$$

$1\leq n\leq 2\times 10^5,1\leq a_i,b_i\leq 2000$

---

1|2解题思路

啊遇到这种题目直接上组合意义 $\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\}\times max\{b_i\})$

---

1|3code

#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\n",(ans+P)*inv[2]%P);
	return 0;
}

__EOF__

本文作者：QuantAsk
本文链接：https://www.cnblogs.com/QuantAsk/p/14828335.html
关于博主：退役OIer，GD划水选手
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！