[BZOJ4036][HAOI2015]按位或（FMT？FWT？）

Address

https://www.lydsy.com/JudgeOnline/problem.php?id=4036

Solution

注：下面的 $\bigotimes$ 和幂，如果运算的对象是序列，则表示或卷积。
（即 $h_k=\sum_{i\text{ OR }j=k}f_i\times g_j$ ）
同时下面 $p[S]$ 表示选数 $S$ 的概率，即题目中的输入。
考虑 $f_{i,S}$ 表示操作了 $i$ 次之后数变成 $S$ 的概率。
那么容易得到 $f_{0,0}=1$ 以及：

$$f_{i,S}=\sum_{A\text{ OR }B=S}f_{i-1,A}\bigotimes p_B$$
根据或卷积的定义，易得 $f_i=p^i$ 。
于是，我们设 $res[S\in[0,2^n)]$ 为变成 $S$ 的期望步数。
那么我们得到：
$$res=\sum_{i=1}^{\infty}i\times (f_i-f_{i-1})$$
$$=-\sum_{i=1}^{\infty}f_i=-\sum_{i=1}^{\infty}p^i$$
我们对 $p$ 求快速莫比乌斯变换 $p'$ 。
转化成：
$$res'[S]=\begin{cases}\frac1{p'[S]-1}&p'[S]\ne 1\\0&p'[S]=1\end{cases}$$
将 $res'$ 求快速莫比乌斯反演后，答案为 $res[2^n-1]$ 。
无解的情况：如果无解，那么对于任意的 $i\in[0,\infty)$ 都满足 $f_{i,2^n-1}=0$ ，这时候就能推出 $res[2^n-1]=0$ 。故当 $res[2^n-1]=0$ 时无解。
复杂度 $O(2^nn)$ 。

Code

代码非常短。

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define For(i, a, b) for (i = a; i <= b; i++)
using namespace std;

const int N = 1048888;
const double eps = 1e-8;
int n, Cm;
double f[N];

void FMT(double *f, int op)
{
    int i, j;
    For (i, 1, n) For (j, 0, Cm)
        if (!((j >> i - 1) & 1))
            f[j | (1 << i - 1)] += f[j] * op;
}

int main()
{
    int i;
    cin >> n;
    Cm = (1 << n) - 1;
    For (i, 0, Cm) scanf("%lf", &f[i]);
    FMT(f, 1);
    For (i, 0, Cm)
        f[i] = fabs(f[i] - 1.0) < eps ? 0 : -1.0 / (1.0 - f[i]);
    FMT(f, -1);
    if (f[Cm] < eps) puts("INF");
    else printf("%.10lf\n", f[Cm]);
    return 0;
}