首先可以将期望转换为$sum_{i=0}^{inf}P(i)$,其中P(i)表示i轮后还没有结束的概率,通过对i轮后的结果容斥,可以得到$P(i)=sum_{j=0}^{2^{n}-2}(sum_{k|j=j}p_{k})^{i}*(-1)^{n-|j|-1}$
对i累加后,发现即$sum_{j=0}^{2^{n}-2}(-1)^{n-|j|-1}sum_{i=0}^{inf}(sum_{k|j=j}p_{k})^{i}=sum_{j=0}^{2^{n}-2}(-1)^{n-|j|-1}*frac{1}{1-sum_{k|j=j}p_{k}}$
$sum_{k|j=j}p_{k}$这个相当于一个n维的前缀和,即FWT的or卷积,然后再累加即可
1 #include<bits/stdc++.h> 2 using namespace std; 3 #define N 2000005 4 int n,m,cnt[N]; 5 double ans,p[N]; 6 int main(){ 7 scanf("%d",&n); 8 m=(1<<n); 9 for(int i=0;i<m;i++)scanf("%lf",&p[i]); 10 for(int i=0;i<n;i++) 11 for(int j=0;j<m;j++) 12 if (j&(1<<i))p[j]+=p[j-(1<<i)]; 13 cnt[0]=(n&1); 14 for(int i=1;i<m;i++)cnt[i]=cnt[i>>1]^(i&1); 15 for(int i=0;i<m-1;i++) 16 if (1-p[i]>1e-8)ans+=(2*cnt[i]-1.0)/(1-p[i]); 17 if (ans<1e-10)printf("INF"); 18 else printf("%.6f",ans); 19 }