2020 CCPC Changchun I Kawaii Courier 题解

最新推荐文章于 2022-05-11 20:55:35 发布

ZLcc_cc

最新推荐文章于 2022-05-11 20:55:35 发布

阅读量198

点赞数 1

分类专栏：随机过程文章标签： acm竞赛数学动态规划

本文链接：https://blog.csdn.net/dadaaklsjdlka/article/details/117453238

版权

随机过程专栏收录该内容

4 篇文章 0 订阅

订阅专栏

I Kawaii Courier

链接: Kawaii Courier.
题意：
给出以

k

为根的

\le10^5

个节点的树。
定义

p_{i,j}

为从点

i

随机游走到点

k

步数恰为

j

的概率。
对于所有

u \in [1, n]

求

u\sum_{i=1}ip_{u,i}x^i

，其中x为给定的数。
题解:
定义

f_u(x)=\sum_{i=1}p_{u,i}x^i

为从点

u

随机游走到

k

点停止的生成函数。
有

u\sum_{i=1}ip_{u,i}x^i=uxf_u'(x)

。
考虑维护 $f_u'(x)$ :
定义

u p (x)

为从节点

x

走到

fa_x

停止的的生成函数。
则

f_u(x)=up_u(x)\prod_{v∈u的祖先,v≠k}up_v(x)

。
求出

u p

后前缀积即是

f

。
关于

u p

的求法以及一阶导数的维护，具体可参考 I Random Walk On Tree。
参考实现：

#include<bits/stdc++.h>

using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
constexpr int mod = 1e9+7;
ll power(ll a, ll b) {ll res = 1;while (b) {if (b & 1)res = res * a % mod;a = a * a % mod;b >>= 1;}return res;}
struct INT{
    int v;
    INT(){v=0;}
    INT(int v):v(v){}
    INT operator+(INT A){return (v+A.v)%mod;}
    INT operator-(INT A){return (v-A.v)%mod;}
    INT operator*(INT A){return (ll)v*A.v%mod;}
    INT operator/(INT A){return v*power(A.v,mod-2)%mod;}
};
struct poly{
    INT f[2];
    poly(){}
    poly(INT a,INT b=0){f[0]=a;f[1]=b;}
    INT& operator[](int i){
        return f[i];
    }
    poly operator+(poly g){
        return {f[0] + g[0], f[1] + g[1]};
    }
    poly operator-(poly g){
        return {f[0] - g[0], f[1] - g[1]};
    }
    poly operator*(poly g){
        return {f[0] * g[0], f[0] * g[1] + f[1] * g[0]};
    }
    poly inv(poly g){
        return {(INT)1 / g[0], g[1] / (g[0] * g[0]) * (-1)};
    }
    poly operator/(poly g){
        return (*this)*inv(g);
    }
};
poly x({1, 1});
vector<int> g[N];
poly up[N];
void dfs_up(int u,int f){
    poly sum_up;
    for(int v:g[u])if(v!=f){
        dfs_up(v,u);
        sum_up=sum_up+up[v];
    }
    up[u]= x / ((poly)g[u].size() - x * sum_up);
}
poly pre_up[N];
void dfs_sum(int u,int f){
    if(f==0){
        pre_up[u]={1,0};
    }else {
        pre_up[u] = pre_up[f] * up[u];
    }
    for(int v:g[u])if(v!=f){
            dfs_sum(v,u);
        }
}

int main() {
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int n,k,p,q;cin>>n>>k>>p>>q;
    x[0]=p*power(q,mod-2)%mod;
    for(int i=1;i<n;i++){
        int u,v;cin>>u>>v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs_up(k,0);
    dfs_sum(k,0);
    int res=0;
    for(int i=1;i<=n;i++){
        res^=((ll)i*(pre_up[i][1].v+mod)%mod*x[0].v%mod);
    }
    cout<<res;
}