莫比乌斯变换(FMT)/子集和变换--luogu3175 [HAOI2015]按位或

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sizeof_you/article/details/84350124

传送门

今天讲子集和变换,其实感觉和高维前缀和差不多

就着这道题学习了一下 F M T FMT

细节可以参考这个博客
简单说就是像 F F T FFT 一样先点值相乘以后再插值回去
变换和反演部分和子集和变换一样
F M T FMT 可以解决一些 F F T FFT 不好解决的奇怪的多项式卷积
像这样:

for(int i=0;i<n;i++)
        for(int j=0;j<ed;j++)
            if(j&(1<<i)) p[j]+=p[j^(1<<i)];

然后这道题就可以用 F M T FMT 来实现
式子大概长这样:
f i , S = x S ( 1 ) S x g i , x f_{i,S}=∑_{x⊆S}(−1)^{|S|−|x|}∗g_{i,x}
完整代码很短,注意判一下无解情况

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define maxn 20
using namespace std;

inline int rd(){
    int x=0,f=1;char c=' ';
    while(c<'0' || c>'9') f=c=='-'?-1:1,c=getchar();
    while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();
    return x*f;
}

const int N=(1<<maxn)+5;
int n,ed;
double p[N],ans;

inline int calc(int x){
    int res=0;
    while(x){
        if(x&1) res++; x>>=1;
    } return res;
}

int main(){
    n=rd(); ed=1<<n;
    for(int i=0;i<ed;i++) scanf("%lf",&p[i]);
    for(int i=0;i<n;i++)
        for(int j=0;j<ed;j++)
            if(j&(1<<i)) p[j]+=p[j^(1<<i)];
    for(int i=0;i<ed-1;i++){
        int x=calc(i); x=n-x;
        if(p[i]>=1.0-1e-12) {puts("INF");return 0;}
        if(x&1) ans+=1.0/(1.0-p[i]);
        else ans-=1.0/(1.0-p[i]);
    }
    printf("%.10lf\n",ans);
    return 0;
}
/*
2
0.25 0.25 0.25 0.25
*/

猜你喜欢

转载自blog.csdn.net/sizeof_you/article/details/84350124