Codeforces Round #302 (Div. 1) D - Road Improvement 树形dp

D - Road Improvemen

思路：0没有逆元！！！！ 不能直接除，要求前缀积和后缀积！！！

#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mk make_pair
#define PII pair<int, int>
#define y1 skldjfskldjg
#define y2 skldfjsklejg

using namespace std;

const int N = 2e5 + ;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int mod = 1e9 + ;

int n;
LL dp[N], ans[N], fdp[N];
vector<int> edge[N];
vector<int> L[N], R[N];

void dfs(int u, int fa) {
    dp[u] = ;
    LL pre = ;
    L[u].resize(edge[u].size());
    R[u].resize(edge[u].size());
    for(int i = ; i < edge[u].size(); i++) {
        int v = edge[u][i];
        L[u][i] = pre;
        if(v == fa) continue;
        dfs(v, u);
        dp[u] = dp[u] * (dp[v] + ) % mod;
        L[u][i] = L[u][i] * (dp[v] + ) % mod;
        pre = L[u][i];
    }

    pre = ;
    for(int i = edge[u].size() - ; i >= ; i--) {
        int v = edge[u][i];
        R[u][i] = pre;
        if(v == fa) continue;
        R[u][i] = R[u][i] * (dp[v] + ) % mod;
        pre = R[u][i];
    }
}

void dfs3(int u, int fa, LL val) {
    ans[u] = dp[u] * (val + ) % mod;
    for(int i = ; i < edge[u].size(); i++) {
        int v = edge[u][i];
        if(v == fa) continue;
        LL nxv = ;
        if(i) nxv = L[u][i - ];
        if(i +  < edge[u].size()) nxv = nxv * R[u][i + ] % mod;
        dfs3(v, u, nxv * (val + ) % mod);
    }
}

int main() {
    scanf("%d", &n);
    for(int i = ; i <= n; i++) {
        int x; scanf("%d", &x);
        edge[i].push_back(x);
        edge[x].push_back(i);
    }
    dfs(, );
    dfs3(, , );
    for(int i = ; i <= n; i++) printf("%lld ", ans[i]);
    return ;
}

/*
*/