洛谷 CF1999F Expected Median 题解

题目传送门

做法

首先， a i ∈ { 0 , 1 } a_i \in \{0, 1\} ai​∈{0,1}，这对我们解题提供了很大的便利。
设 $cn{t}_0$ 为 $0$ 出现的个数。我们发现，一个序列的中位数为
$$\{\begin{array}{ll}0& \text{if }cn{t}_0\ge \frac{k+1}{2}\\ 1& \text{if }cn{t}_0<\frac{k+1}{2}\end{array}$$
所以题目相当于是问中位数 $=1$ 的 $a$ 的子序列有多少个。

假设我们知道了 $cnt0$，那么怎么算中位数 $=1$ 的 $a$ 的子序列有多少个呢。
既然是方案数，那我们考虑组合计数。

首先，如果 $cn{t}_0>=\frac{k+1}{2}$，那么方案数就是 $0$，否则就是选择 $0$ 的方案数 $\times$ 选择 $1$ 的方案数。

选择 $1$ 的数量 $cn{t}_1=k-cn{t}_0$。
设 $0$ 的数量为 $ma{x}_0$，$1$ 的数量为 $ma{x}_1$。

$0$ 的选择方案数量为 $\left(\genfrac{}{}{0ex}{}{ma{x}_0}{cn{t}_0}\right)$，$1$ 的选择方案为 $\left(\genfrac{}{}{0ex}{}{ma{x}_1}{cn{t}_1}\right)$

则总方案数为 $\left(\genfrac{}{}{0ex}{}{ma{x}_0}{cn{t}_0}\right)\times \left(\genfrac{}{}{0ex}{}{ma{x}_1}{cn{t}_1}\right)$

那 $cn{t}_0$ 怎么知道？枚举呗。
最终答案为
$\sum _{cn{t}_0}^{max0}\left(\genfrac{}{}{0ex}{}{ma{x}_0}{cn{t}_0}\right)\times \left(\genfrac{}{}{0ex}{}{ma{x}_1}{k-cn{t}_0}\right)[cn{t}_0<k-cn{t}_0]$

实现

组合数按公式
预处理就行，预处理阶乘和阶乘的逆元，不知道什么是逆元的点击查看。

然后直接求答案即可

代码

#include <iostream>
#include <cstring>
#include <algorithm>
 
using namespace std;
 
typedef long long LL;
const int N = 200010, mod = 1e9 + 7;
 
int n, k;
int a[N];
int fact[N], infact[N];
 
int qmi(int a, int k)
{
	int ans = 1;
	while (k)
	{
		if (k & 1) ans = (LL)ans * a % mod;
		a = (LL)a * a % mod;
		k >>= 1;
	}
	return ans;
}
 
void init(int n)
{
	fact[0] = infact[0] = 1;
	for (int i = 1; i <= n; i ++ )
	{
		fact[i] = (LL)fact[i - 1] * i % mod;
		infact[i] = (LL)infact[i - 1] * qmi(i, mod - 2) % mod;
	}
}
 
int C(int a, int b)
{
	if (a < b) return 0;
	return (LL)fact[a] * infact[b] % mod * infact[a - b] % mod;
}
 
int main()
{
	int T;
	scanf("%d", &T);
	init(N - 5);
	while (T -- )
	{
		scanf("%d%d", &n, &k);
		int max0 = 0, max1 = 0;
		for (int i = 1; i <= n; i ++ )
		{
			scanf("%d", &a[i]);
			if (a[i] == 0) max0 ++ ;
			else max1 ++ ;
		}
		
		int sum = 0;
		for (int cnt0 = 0; cnt0 <= max0; cnt0 ++ )
			if (cnt0 < (k + 1) / 2)
				sum = (sum + (LL)C(max0, cnt0) * C(max1, k - cnt0) % mod) % mod;
		
		printf("%d\n", sum);
	}
	
	return 0;
}