Loj #2542 随机游走

Solution

套路题目。
首先求的是全部都被覆盖的期望，那么根据$\text{min-max}$容斥，我们可以计算某一个点第一次被染的期望。那么现在问题变成了给定一个点集以及一个起点，求期望下多久第一次走到点集中的某一个点。
明显，对于一个给定的点集$S$，我们设$f[i]$表示在$i$号点的期望，那么可以高消。考试时候写这个也可以过。
当然这个就没有用到树的性质。我们还是设$f[i]$表示期望。那么有

$$\begin{cases} f[i]= 0 (i \in S)\\ f[i]= \frac{1}{d_i}(f[fa] + \sum f[sons]) + 1(\text{other wise})\\ \end{cases}$$

其中$d_i$表示点$i$的度数。那么我们对于一个$f[i]$，可以只用其父亲来表示，然后这个高消就可以$O(n)$了。

Code

#include <bits/stdc++.h>
#define mod 998244353
using namespace std;
const int N = 18;
void pd(int &x){if (x >= mod) x -= mod;else if (x < 0) x += mod;}
int qpow(int a,int b,int p = mod)
{
    int res = 1;
    while (b)
    {
        if (b & 1) res = 1ll * res * a % p;
        a = 1ll * a * a % p; b >>= 1;
    }
    return res;
}

int inv(int a){return qpow(a , mod-2 , mod);}


struct E{
    int to , nex;
}e[N*2+10];int head[N+10] , ecnt;
void adde(int fr,int to){e[++ecnt]=(E){to , head[fr]}; head[fr] = ecnt;}
void addedge(int fr,int to){adde(fr,to);adde(to,fr);}

int E[1<<N] ;
int A[N+10] , B[N+10] , dg[N+10];

int n , q , rt , ans;


void solve(int o,int fa,int mask)
{
    A[o] = B[o] = 0; 
    if ( ( 1 << (o - 1) ) & mask ) return;

    int suma = 0 , sumb = 0 , coef = 0;

    for (int j = head[o];j;j=e[j].nex)
    if (e[j].to != fa)
    {
        solve(e[j].to , o , mask);
        pd( suma += A[e[j].to] );
        pd( sumb += B[e[j].to] );
    }

    int inv = ::inv(dg[o]);

    coef = 1ll * (1 - 1ll * suma * inv % mod + mod) % mod;
    coef = ::inv(coef);
    A[o] = 1ll * inv * coef % mod;
    B[o] = 1ll * (1 + 1ll * sumb * inv %mod ) * coef % mod;
    if (o == rt)
    {
        E[mask] = B[rt];  
    }
}

int main()
{
    scanf("%d%d%d",&n,&q,&rt);
    for (int i = 1;i < n;i++)
    {
        int u , v;scanf("%d%d" , &u , &v);
        addedge(u , v);
        dg[u]++;dg[v]++;
    }
    int S = 1 << n;
    for (int mask = 0;mask < S;mask++)
    {
        solve(rt , 0 , mask);
    }

    while (q--)
    {
        int mask = 0 , k;ans = 0;
        scanf("%d",&k);
        for (int i = 1;i <= k;i++) {int u;scanf("%d" , &u); mask |= 1 << (u - 1);}
        for (int sub = mask ;sub; sub = (sub - 1) & mask)
        pd(ans += __builtin_popcount(sub) & 1 ? E[sub] : -E[sub]);
        printf("%d\n",ans);
    }
    return 0;
}