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

最新推荐文章于 2019-08-10 20:38:00 发布

xyz32768

最新推荐文章于 2019-08-10 20:38:00 发布

阅读量266

点赞数

分类专栏： BZOJ、UOJ、LOJ 文章标签： FMT FWT

本文链接：https://blog.csdn.net/xyz32768/article/details/82227894

版权

BZOJ、UOJ、LOJ 专栏收录该内容

341 篇文章 1 订阅

订阅专栏

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 OR B = S f i - 1, A ⨂ p B

$f_{i,S}=\sum_{A\text{ OR }B=S}f_{i-1,A}\bigotimes p_B$
根据或卷积的定义，易得

fi=pi f i = p i $f_i=p^i$ 。
于是，我们设

res[S∈[0,2n)] r e s [ S ∈ [ 0 , 2 n ) ] $res[S\in[0,2^n)]$ 为变成

S S $S$ 的期望步数。
那么我们得到：

r e s = \sum_{i = 1}^{\infty} i \times (f_{i} - f_{i - 1})

$res=\sum_{i=1}^{\infty}i\times (f_i-f_{i-1})$

= - \sum i = 1 \infty f i = - \sum i = 1 \infty p i

$=-\sum_{i=1}^{\infty}f_i=-\sum_{i=1}^{\infty}p^i$
我们对

p p $p$ 求快速莫比乌斯变换

p^{'}

$p'$ 。
转化成：

r e s' [S] = {1 p ' [ S ] - 1 0 p' [S] \neq 1 p' [S] = 1

$res'[S]=\begin{cases}\frac1{p'[S]-1}&p'[S]\ne 1\\0&p'[S]=1\end{cases}$
将

res′ r e s ′ $res'$ 求快速莫比乌斯反演后，答案为

res[2n−1] r e s [ 2 n − 1 ] $res[2^n-1]$ 。
无解的情况：如果无解，那么对于任意的

i∈[0,∞) i ∈ [ 0 , ∞ ) $i\in[0,\infty)$ 都满足

fi,2n−1=0 f i , 2 n − 1 = 0 $f_{i,2^n-1}=0$ ，这时候就能推出

res[2n−1]=0 r e s [ 2 n − 1 ] = 0 $res[2^n-1]=0$ 。故当

res[2n−1]=0 r e s [ 2 n − 1 ] = 0 $res[2^n-1]=0$ 时无解。
复杂度

O(2nn) O ( 2 n n ) $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;
}