Solution -「AGC 019F」「AT 2705」Yes or No

\(\mathcal{Description}\)

  Link.

  有 \(n+m\) 个问题，其中 \(n\) 个答案为 yes，\(m\) 个答案为 no。每次你需要回答一个问题，然后得知这个问题的正确答案。求最优策略下期望答对的题数。

  \(n,m\le5\times10^5\)。

\(\mathcal{Solution}\)

  显然贪心策略：当 \(n\not=m\)，猜较多的答案。

  设 \(n>m\)，无脑猜 yes，就一定能答对 \(n\) 道题。那么所有 \(n\not=m\) 的情形下，猜对的期望次数之和就是 \(n\)。接下来只需要考虑 \(n=m\) 时的贡献。

  其实 OEIS 能找到 www。首先，\(n=m\) 时，猜中的概率显然为 \(\frac{1}2\)。那么贡献次数呢？考虑成一张为网格图，从 \((0,0)\) 向下或向右走，走到 \((n,m)\)，求进过 \((i,i)\) 的概率。这就是经典的组合数问题嘛。综上，答案为：

\[\max\{n,m\}+\frac{1}{2\binom{n+m}{n}}\sum_{i=1}^{\min\{n,m\}}\binom{2i}{i}\binom{n+m-2i}{n-i} \]

  求出来就好啦，复杂度 \(\mathcal O(n+m)\)。

\(\mathcal{Code}\)

#include <cstdio>

const int MAXN = 5e5, MOD = 998244353;
int n, m, fac[MAXN * 2 + 5], ifac[MAXN * 2 + 5];

inline int qkpow ( int a, int b, const int p = MOD ) {
	int ret = 1;
	for ( ; b; a = 1ll * a * a % p, b >>= 1 ) ret = 1ll * ret * ( b & 1 ? a : 1 ) % p;
	return ret;
}

inline void init ( const int n ) {
	fac[0] = 1;
	for ( int i = 1; i <= n; ++ i ) fac[i] = 1ll * i * fac[i - 1] % MOD;
	ifac[n] = qkpow ( fac[n], MOD - 2 );
	for ( int i = n - 1; ~ i; -- i ) ifac[i] = ( i + 1ll ) * ifac[i + 1] % MOD;
}

inline int C ( const int n, const int m ) {
	return n < m ? 0 : 1ll * fac[n] * ifac[m] % MOD * ifac[n - m] % MOD;
}

int main () {
	scanf ( "%d %d", &n, &m );
	if ( n < m ) n ^= m ^= n ^= m;
	init ( n + m );
	int ans = n, inv = ( MOD + 1ll ) / 2 * qkpow ( C ( n + m, n ), MOD - 2 ) % MOD;
	for ( int i = 1; i <= m; ++ i ) {
		ans = ( ans + 1ll * inv * C ( 2 * i, i ) % MOD * C ( n + m - 2 * i, n - i ) ) % MOD;
	}
	printf ( "%d\n", ans );
	return 0;
}