[FMT][DP] BZOJ 4036: [HAOI2015]按位或

最新推荐文章于 2024-08-02 23:39:45 发布

Vectorxj

最新推荐文章于 2024-08-02 23:39:45 发布

阅读量422

点赞数 2

分类专栏： DFT 动态规划

本文链接：https://blog.csdn.net/Vectorxj/article/details/78254658

版权

动态规划同时被 2 个专栏收录

62 篇文章 0 订阅

订阅专栏

DFT

20 篇文章 0 订阅

订阅专栏

$Description$

刚开始有一个集合为空集，全集为 $U$ 。
每次以 $P_T$ 的概率给一个集合 $T$ ，使拥有的那个集合变成他与这个集合的并。
问变成全集 $U$ 的期望步数。

$Solution$

设 $f_S$ 为 $P_S$ 的集合幂级数。他的莫比乌斯变换

f^S = \sum T \subseteq S f T

$\hat{f}_S=\sum_{T\subseteq S}f_T$ 同时有莫比乌斯反演

f S = \sum T \subseteq S (- 1) | T | - | S | f^T

$f_S=\sum_{T\subseteq S}(-1)^{|T|-|S|}\hat{f}_T$ 题目要求的就是集合并卷积：

h S = \sum L \subseteq U \sum R \subseteq U f L g R [L \cup R = S]

$h_S=\sum_{L\subseteq U}\sum_{R\subseteq U}f_Lg_R[L\cup R=S]$ 通过莫比乌斯变换：

h^S = = = = \sum T \subseteq S h T \sum L \subseteq S \sum R \subseteq S f L g R (\sum L \subseteq S f L) (\sum R \subseteq S g R) f^S g^S

$\begin{eqnarray} \hat{h}_S&=&\sum_{T\subseteq S}h_T\\ &=&\sum_{L\subseteq S}\sum_{R\subseteq S}f_Lg_R\\ &=&(\sum_{L\subseteq S}f_L)(\sum_{R\subseteq S}g_R)\\ &=&\hat{f}_S\hat{g}_S \end{eqnarray}$ 所以直接

FMT $\text{FMT}$ 计算出相应的莫比乌斯变换，然后莫比乌斯反演

FMT−1 $\text{FMT}^{-1}$ 回去就好了。
时间复杂度是

O(2nn) $O(2^nn)$ 的。

using namespace std;

const int N = 22;
const int eps = 1e-8;

int n, m;
double p[1 << N];

inline void FMT(double *a, int n, int f) {
    int m = 1 << n;
    for (int i = 0; i < n; i++)
        for (int S = 1; S < m; S++)
            if ((S >> i) & 1)
                a[S] += f * a[S ^ (1 << i)];
}

int main(void) {
    freopen("1.in", "r", stdin);
    scanf("%d", &n); m = 1 << n;
    for (int i = 0; i < m; i++)
        scanf("%lf", &p[i]);
    FMT(p, n, 1);
    for (int i = 0; i < m; i++)
        if (fabs(1 - p[i]) <= eps) p[i] = 0;
        else p[i] = 1 / (p[i] - 1);
    FMT(p, n, -1);
    if (fabs(p[m - 1]) <= eps) printf("INF\n");
    else printf("%.10lf\n", p[m - 1]);
}