BZOJ 4036 [HAOI2015]按位或

快速莫比乌斯变换

关于集合幂级数的更多内容请看2015年集训队论文最后一篇VFK的。orz orz

主要思路就是把求一个集合幂级数的卷积转化成一个求子集和的问题。过程是作出莫比乌斯变换，变成求和，再快速莫比乌斯反演回去。

记 $f_i(s)$ 为 $i$ 次过后，并集为 $s$ 的概率。记 $h(s)$ 表示最终并集为 $s$ 的期望次数，记$U = 2^n-1$

那我们要求的就是

$h(U) = \sum _{i=1} ^{\infty} i*[f_i(U) - f_{i-1}(U) ]$

对两边做莫比乌斯变换

$\hat h(U) = \sum _{i=1} ^{\infty}i * [\hat f_i(U) - \hat f_{i-1}(U) ]$

我们希望快速求出右边，从而反演出左边，观察 $\hat f_i(U)$ 的定义式

$\hat f_i(U) = \sum _{s_1 \in U} f_{i-1}(s_1) * \sum _{s_2 \in U} f_1(s_2)$

整合一下，有

$\hat f_i(U) = [\hat f_1i(U)] ^ i$

因此，有

$\hat h(U) = \sum _{i=1} ^{\infty} i*[\hat f_1(U)]^i - [\hat f_{1}(U)]^{i-1} ]$

右边的和式可以通过扰动法求出等比数列的和，显然是收敛的，即

$\hat h(U) = \frac {1} {\hat f_1(U) - 1}$

把 $\hat h(U)$ 反演回 $h(U)$ 即可得到答案

$h(U) = \sum_{s \in U} \frac{1}{\hat f_1(s)-1} * (-1) ^ {U中1的个数-s中1的个数}$

一个细节是上述的等比数列求和时，若$f_1(U)=1$时不能用直那样求和，否则会除以$0$ . 正确做法是忽略它，因为它的和总是$1-1=0$

#include<cstdio>
#define N 21
using namespace std;
namespace runzhe2000
{
    int n, g[1<<N]; double f[1<<N], h;
    void main()
    {
        scanf("%d",&n); 
        int U = (1<<n)-1, V;
        g[0] = 1;
        for(int i = 0; i <= U; i++) 
        {
            scanf("%lf",&f[i]);
            i?g[i] = -g[i-(i&-i)]:0;
            f[i]?V|=i:0;
        }
        if(V!=U){puts("INF");return;}
        for(int i = 0; i < n; i++)
            for(int j = 1; j <= U; j++) 
                (j&(1<<i)) ? f[j] += f[j-(1<<i)] : 0;
        for(int i = 0; i < U; i++) h += g[i] / (f[i] - 1);
        printf("%.8lf\n",h);    
    }
}
int main()
{
    runzhe2000::main();
}