【AtCoder】【DP】【组合数学】BBQ Hard(AGC001)

11 篇文章 0 订阅
7 篇文章 0 订阅

题意:

有n个包,一个包里面有一根竹签,上面有编号i,还有Ai个A物品,Bi个B物品。现在选择两个包,用两个竹签将A物品和B物品串起来。两种方法是不一样的,当且仅当选择的竹签的编号不同(忽略顺序)或者A,B物品的摆放顺序不同(可重复排列)。
下面是N=3的情况:
在这里插入图片描述

数据范围:

2≦N≦200,000
1≦Ai≦2000,1≦Bi≦2000

思路:

考试的时候只会骗…首先,很容易想到一个O(n^2)的算法,也就是枚举两个包,然后算这两个包的可重复排列,就是一道大版题了…这样子可以骗60分(够了够了 )。
其实我们通常在使用组合数的时候,都忽略了组合数的基本定义。这道题的答案可以是:
∑ i = 1 N ∑ j = i + 1 N C A i + B i + A j + B j A i + A j \sum_{i=1}^{N}\sum_{j=i+1}^{N}C_{A_i+B_i+A_j+B_j}^{A_i+A_j} i=1Nj=i+1NCAi+Bi+Aj+BjAi+Aj
然后可以转化为一个经典组合问题:从(-Ai,-Bi)到(Aj,Bj),只能往右或往上走的方案数。
这样就将每一个(-Ai,-Bi)的dp值初始化为1,然后从左下向右上递推就可以了。
还需要注意的是,这样算出来还需要去重。
A n s = ∑ i = 1 N ∑ j = i + 1 N C A i + B i + A j + B j A i + A j Ans=\sum_{i=1}^{N}\sum_{j=i+1}^{N}C_{A_i+B_i+A_j+B_j}^{A_i+A_j} Ans=i=1Nj=i+1NCAi+Bi+Aj+BjAi+Aj
= ( ∑ i = 1 N ∑ j = 1 N C A i + B i + A j + B j A i + A j − ∑ i = 1 N C 2 A i + 2 B i A i + B i ) 2 =\dfrac{(\sum_{i=1}^{N}\sum_{j=1}^{N}C_{A_i+B_i+A_j+B_j}^{A_i+A_j}-\sum_{i=1}^{N}C_{2A_i+2B_i}^{A_i+B_i})}{2} =2(i=1Nj=1NCAi+Bi+Aj+BjAi+Aji=1NC2Ai+2BiAi+Bi)
然后就可以计算答案了。
其实我觉得这道题的正解比较难想,感觉是出题人想到可以转换这个模型之后,才硬搞出来的这样一道题目。当然这只是个人的看法,反正我在看到正解的做法的时候是很震惊的…

#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXN 200000
#define MAXV 4000
#define MO 1000000007
using namespace std;
int n,A[MAXN+5],B[MAXN+5];
int dp[MAXV+5][MAXV+5];
int fact[MAXV*2+5],inv[MAXV*2+5];
int PowMod(int a,int b)
{
	int ret=1;
	while(b)
	{
		if(b&1)
			ret=1LL*ret*a%MO;
		a=1LL*a*a%MO;
		b>>=1;
	}
	return ret;
}
void prepare()
{
	fact[0]=1;
	for(int i=1;i<=MAXV*2;i++)
		fact[i]=1LL*fact[i-1]*i%MO;
	inv[MAXV*2]=PowMod(fact[MAXV*2],MO-2);
	for(int i=MAXV*2-1;i>=0;i--)
		inv[i]=1LL*inv[i+1]*(1LL*i+1LL)%MO;
}
int C(int a,int b)
{
	return 1LL*fact[a]*inv[b]%MO*inv[a-b]%MO;
}
int main()
{
//	freopen("bbq.in","r",stdin);
//	freopen("bbq.out","w",stdout);
	prepare();
	scanf("%d",&n);
	int maxval=-1;
	for(int i=1;i<=n;i++)
	{
		scanf("%d %d",&A[i],&B[i]);
		maxval=max(maxval,A[i]);
		maxval=max(maxval,B[i]);
	}
	maxval++;
	for(int i=1;i<=n;i++)
		dp[maxval-A[i]][maxval-B[i]]++;
	for(int i=-maxval+1;i<=maxval;i++)
		for(int j=-maxval+1;j<=maxval;j++)
			dp[i+maxval][j+maxval]=(1LL*dp[i+maxval][j+maxval]+1LL*dp[i-1+maxval][j+maxval]+1LL*dp[i+maxval][j-1+maxval])%MO;
	int ans=0;
	for(int i=1;i<=n;i++)
	{
		ans=(1LL*ans+1LL*dp[A[i]+maxval][B[i]+maxval])%MO;
		ans=((1LL*ans-1LL*C(A[i]+B[i]+A[i]+B[i],A[i]+A[i]))%MO+MO)%MO;
	}
	ans=1LL*ans*PowMod(2,MO-2)%MO;
	printf("%d\n",ans);
	return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值