AtCoder Grand Contest 019F Yes or No 组合数学

最新推荐文章于 2021-09-23 18:40:22 发布

olahiuj

最新推荐文章于 2021-09-23 18:40:22 发布

阅读量241

点赞数

分类专栏： AtCoder c++ 组合数学文章标签： AtCoder

本文链接：https://blog.csdn.net/jpwang8/article/details/88893900

版权

c++ 同时被 3 个专栏收录

1073 篇文章 0 订阅

订阅专栏

AtCoder

36 篇文章 0 订阅

订阅专栏

组合数学

6 篇文章 0 订阅

订阅专栏

Description

有n+m道题目，n道题目的答案为Y，m道为N
现在你要回答这n+m道题目，每次会从没答过的题目中抽一道回答，若回答正确就加一分。问最优策略下得分的期望

Solution

tourist的题目真是劲啊

我们可以用一对整数对(x,y)表示一个剩余x个1问题，剩余y个2问题的局面。显然每一步的最优策略就是猜剩得多的那个，这样等价于我们在每个点有概率往两个方向走一步，且只有一种方向会让正确数+1

在这里插入图片描述
图是随便抄来的，我们走过图中蓝色的边就会使得答案+1
不难发现无论如何从(n,m)走到(1,1)一定会走过max(n,m)条蓝边

再考虑在(i,i)这样局面的情况下要怎么走，显然我们只能是随便猜一个答案。那么问题就变成了多少条路径会经过这个点，这些走法都会使得答案加上1/2

于是答案就是 $\frac{\max(n,m)\binom{n+m}{n}+\frac{1}{2}\sum_{i=1}^{\min(n,m)}\binom{2i}{i}\binom{n+m-2i}{n-i}}{\binom{n+m}{n}}$

Code

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

typedef long long LL;
const int MOD=998244353;
const int N=2000005;

LL fac[N],inv[N];

void upd(LL &x,LL v) {
	x+=v; (x>=MOD)?(x-=MOD):0;
}

void pre() {
	fac[0]=fac[1]=inv[0]=inv[1]=1;
	rep(i,2,N-1) {
		fac[i]=fac[i-1]*i%MOD;
		inv[i]=(MOD-MOD/i)*inv[MOD%i]%MOD;
	}
	rep(i,2,N-1) inv[i]=inv[i]*inv[i-1]%MOD;
}

LL ksm(LL x,LL dep) {
	LL res=1;
	for (;dep;dep>>=1) {
		(dep&1)?(res=res*x%MOD):0;
		x=x*x%MOD;
	}
	return res;
}

LL C(int n,int m) {
	return fac[n]*inv[m]%MOD*inv[n-m]%MOD;
}

int main(void) {
	pre(); int n,m;
	scanf("%d%d",&n,&m);
	if (n>m) std:: swap(n,m);
	LL ans=0;
	rep(i,1,n) upd(ans,C(i+i,i)*C(n+m-i*2,n-i)%MOD);
	ans=ans*ksm(C(n+m,n)*2%MOD,MOD-2)%MOD;
	upd(ans,m); printf("%lld\n", ans);
	return 0;
}