BZOJ4036: [HAOI2015]按位或

vfk论文上的题

记$U=2^n-1$，因为$\forall_{T\subseteq S}U|T=U$
我们把p看成集合幂级数，那么有
$ans=\sum_{k=1}^{∞}k*(p^k_U-p^{k-1}_U)$
若我们定义集合幂级数f的第T项为$f_T=\sum_{k=1}^{∞}k*(p^k_T-p^{k-1}_T)$
将两边同时做莫比乌斯变换（子集和变换）
$F_T=\sum_{k=1}^{∞}k*(P^k_T-P^{k-1}_T)$
发现右边是个等比数列求和的东西，当$P_T=1$时$F_T=0$，否则$F_T=\frac{1}{1-P_T}$
求出$F$的每一项后做一次莫比乌斯逆变换求出$f_U$

code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;

const int maxn = 1<<20;
const double eps = 1e-9;

int n,N;
double f[maxn];

void FWT(double s[],int sig)
{
    for(int i=0;i<n;i++)
    {
        int t=1<<i;
        for(int j=0;j<N;j++) if(!(j>>i&1))
            f[j+t]+=sig*f[j];
    }
}

int main()
{
    //freopen("tmp.in","r",stdin);
    //freopen("tmp.out","w",stdout);

    scanf("%d",&n); N=1<<n;
    for(int i=0;i<N;i++) scanf("%lf",&f[i]);
    FWT(f,1);
    for(int i=0;i<N;i++) f[i]=fabs(f[i]-1)<eps?0.0:-1.0/(1.0-f[i]);
    FWT(f,-1);
    if(fabs(f[N-1])<eps) puts("INF");
    else printf("%.8lf\n",f[N-1]);

    return 0;
}