【LOJ】#2127. 「HAOI2015」按位或
题解
听说这是一道论文题orz
\(\sum_{k = 1}^{\infty} k(p^{k} - p^{k - 1})\)
答案是这个多项式的第\(2^N - 1\)项的系数
我们反演一下,卷积变点积
\(\hat{f_{S}} = \sum_{k = 1}^{\infty} k(\hat{p_{S}}^{k} - \hat{p_{S}}^{k - 1})\)
这是个等比数列啊,怎么推呢= =
设答案为\(S\),如果我在相邻的两项之间
例如\(2(\hat{p_{S}}^{2} - \hat{p_{S}}^{1})\)
\((\hat{p_{S}}^{1} - \hat{p_{S}}^{0})\)每项多加一个\(\hat{p_{S}}^{k}\)再减去
最后会有一个\(\infty \hat{p}^{\infty} - \hat{p_{S}}^{0}\)
所以
\(S = \infty \hat{p}^{\infty} - \sum_{k = 0}^{\infty} \hat{p}^{k}\)
\(\hat{p}S = \infty \hat{p}^{\infty} - \sum_{k = 1}^{\infty} \hat{p}^{k}\)
上式减下式
\((1 - \hat{p})S = -1\)
\(S = - \frac{1}{1 - \hat{p}}\)
所以就有
\(\hat{f} = \left\{\begin{matrix}
-\frac{1}{1 - \hat{p}} & \hat{p} < 1\\
0 & \hat{p} = 1
\end{matrix}\right.\)
最后把F反演回去就行
代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <vector>
//#define ivorysi
#define MAXN 2000005
#define eps 1e-8
#define mo 974711
#define pb push_back
#define mp make_pair
#define pii pair<int,int>
#define fi first
#define se second
using namespace std;
typedef long long int64;
typedef unsigned int u32;
typedef double db;
const int64 MOD = 998244353;
int N,L;
db P[MAXN],F[MAXN];
bool dcmp(db a,db b) {
return fabs(a - b) < eps;
}
template <class T>
void FMT(T *a,T ty) {
for(int i = 1 ; i < L ; i <<= 1) {
for(int j = 0 ; j < L ; ++j) {
if(j & i) {
a[j] = a[j] + ty * a[j ^ i];
}
}
}
}
int main() {
#ifdef ivorysi
freopen("f1.in","r",stdin);
#endif
scanf("%d",&N);
L = 1 << N;
for(int i = 0 ; i < L ; ++i) scanf("%lf",&P[i]);
FMT(P,1.0);
for(int i = 0 ; i < L ; ++i) {
if(dcmp(1.0,P[i])) F[i] = 0;
else F[i] = -1/(1 - P[i]);
}
FMT(F,-1.0);
if(dcmp(F[L - 1],0)) puts("INF");
else printf("%.6lf\n",F[L - 1]);
return 0;
}