Loading

题解「AT1983 [AGC001E] BBQ Hard」

转载注明来源:https://www.cnblogs.com/syc233/p/13627377.html


这题的模型转化挺巧妙的,不过也都是套路。


套路:从棋盘的 \((0,0)\) 走到 \((n,m)\) ,每步只能向上或向右走的方案数为 \({n+m \choose n}\)

考虑一个DP:令 \(f_{i,j}\) 表示从 \((0,0)\) 走到 \((i,j)\) 的方案数,有转移方程:

\[f_{i,j}=f_{i-1,j}+f_{i,j-1} \]

尝试用组合数替换一下,得:

\[{i+j \choose i}={i+j-1 \choose i-1}+{i+j+1 \choose i} \]

这恰好是杨辉三角,说明 \(f_{i,j}={i+j \choose i}\)


回到这道题。

由上面的套路,不难想出一个解法:枚举 \(i,j\) ,求出 \((0,0)\) 走到 \((a_i+a_j,b_i+b_j)\) 的方案数。

然而这和暴力没什么两样,都是 \(O(n^2)\) 级别的,因为终点与 \(i,j\) 都有关。

尝试平移下棋盘,则变成了求 \((-a_i,-b_i)\) 走到 \((a_j,b_j)\) 的方案数。

这样就好统计了。将每一个 \(f_{-a_i,-b_i}\) 赋初值 \(1\) ,DP统计答案即可。

因为题目中 \(i \not =j\) ,还要减去 \(i=j\) 的情况,即从 \((-a_i,-b_i)\) 走到 \((a_i,b_i)\) 的方案数,即 \({2a_i+2b_i \choose 2a_i}\)

因为每一对 \((i,j)\) 只算一次,所以最后答案还要除以 \(2\)

因为是模意义下除以 \(2\) ,所以应该乘上 \(2\)\(10^9+7\) 意义下的逆元。


\(\text{Code}:\)

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#define maxn 200005
#define maxm 2001
#define Rint register int
#define INF 0x3f3f3f3f
using namespace std;
typedef long long lxl;
const lxl mod=1e9+7;

template <typename T>
inline void read(T &x)
{
	x=0;T f=1;char ch=getchar();
	while(ch<'0'||ch>'9') {if(ch=='-') f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	x*=f;
}

int N,A[maxn],B[maxn];
lxl f[(maxm<<1)+5][(maxm<<1)+5],ans;
lxl inv[(maxm<<2)+5],fac[(maxm<<2)+5],inv2[(maxm<<2)+5];

inline void init()
{
	inv[0]=inv[1]=fac[0]=fac[1]=inv2[0]=inv2[1]=1;
	for(int i=2;i<=(maxm<<2);++i)
	{
		inv[i]=(mod-mod/i)*inv[mod%i]%mod;
		fac[i]=fac[i-1]*i%mod;
		inv2[i]=inv2[i-1]*inv[i]%mod;
	}
}

inline lxl C(int n,int m)
{
	return fac[n]*inv2[m]%mod*inv2[n-m]%mod;
}

int main()
{
	// freopen("AT1983.in","r",stdin);
	read(N);
	init();
	for(int i=1;i<=N;++i)
	{
		read(A[i]),read(B[i]);
		++f[maxm-A[i]][maxm-B[i]];
	}
	for(int i=1;i<=(maxm<<1);++i)
		for(int j=1;j<=(maxm<<1);++j)
			(f[i][j]+=(f[i-1][j]+f[i][j-1])%mod)%=mod;
	for(int i=1;i<=N;++i)
	{
		(ans+=f[maxm+A[i]][maxm+B[i]])%=mod;
		(ans+=mod-C(2*A[i]+2*B[i],2*A[i]))%=mod;
	}
	printf("%lld\n",ans*inv[2]%mod);
	return 0;
}

posted @ 2020-09-07 15:55  GoPoux  阅读(131)  评论(0编辑  收藏  举报