其他分享
首页 > 其他分享> > 【luogu P3175】按位或(min-max容斥)(高维前缀和 / FWT)

【luogu P3175】按位或(min-max容斥)(高维前缀和 / FWT)

作者:互联网

按位或

题目链接:luogu P3175

题目大意

有一个数 0 你一开始,然后每次你可以与上一个数 0~2^n-1 中的,每个数有它被你选择的概率。
然后问你期望要弄多少次才能使得这个数变成 2^n-1。

思路

首先这个弄成 \(2^n-1\) 显然不好弄,我们考虑一个神奇的东西,就是 min-max 容斥。
因为这个 \(\min,\max\) 它不一定要是最大值最小值,它可以是最早出现最晚出现之类的。
所以我们可以视作最后一次操作让 \(2^n-1\) 完成是最晚出现,那最早出现就是第一次开始拼 \(2^n-1\)。
那再看回去 min-max 容斥的式子:\(\max(T)=\sum\limits_{T\subset S}(-1)^{|S|}\min(S)\)

那我们就要求出每个子集第一次被覆盖到的期望时间,设为 \(f(S)\)。
考虑生成函数,对于每次如果覆盖到就结束,没有覆盖到就要继续,然后没有覆盖到相当于覆盖了 \(S\) 的补集(这里设为 \(nS\))的子集,然后设一次操作选的集合是 \(S\) 的子集的概率为 \(P(S)\)。
\(f(S)=P(nS)f(S)+1\)
\(f(S)=\dfrac{1}{1-P(nS)}\)

然后再看怎么求 \(P(S)\),这个其实简单,直接一个高位前缀和就好了。
不过后来发现也可以用 FWT 之类的。

就是好像是对于 FWT 的或,它是这样的:
\(C_i=\sum\limits_{j|k=i}A_jB_k\)
然后如果设 \(f(C_i)=\sum\limits_{j\subseteq i}C_j\)
那 \(f(C_i)=\sum\limits_{j|k\subseteq i}A_jB_k=\sum\limits_{j\subseteq i,k\subseteq i}A_jB_k=f(A_i)f(B_i)\)

那 FWT 不就是跟 FFT 差不多弄个变换然后逆变换吗?
那我们 FWT 或变换得到的就是子集和!

很神奇吧。

然后又一些细节,就是记得有个东西是如果无解输出 INF。
那首先如果答案是 \(0\) 你得 INF,而且可能因为你 \(f(S)\) 的求里面可能下面是 \(0\)(就 \(P(nS)=1\)),所以碰到这些你要直接跳过,不然会出问题。

代码

#include<cstdio>

using namespace std;

const int N = 20;
const double eps = 1e-8;
int n, xs[1 << N];
double p[1 << N], f[1 << N], ans;

int main() {
	scanf("%d", &n);
	for (int i = 0; i < (1 << n); i++) scanf("%lf", &p[i]);
	
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < (1 << n); j++)
			if ((j >> i) & 1) p[j] += p[j ^ (1 << i)];
	}
	
	xs[0] = -1;
	for (int i = 1; i < (1 << n); i++) {
		xs[i] = -xs[i ^ (i & (-i))];
		if (1 - p[i ^ ((1 << n) - 1)] < eps) continue;
		f[i] = 1.0 / (1 - p[i ^ ((1 << n) - 1)]);
		ans += f[i] * xs[i];
	}
	if (ans < eps) printf("INF");
		else printf("%.8lf", ans);
	
	return 0;
}

标签:limits,min,luogu,sum,容斥,max,FWT,subseteq
来源: https://www.cnblogs.com/Sakura-TJH/p/luogu_P3175.html