Codeforces Bubble Cup 8 - Finals [Online Mirror] B. Bribes lca

题目链接：

http://codeforces.com/contest/575/problem/B

题解：

把链u,v拆成u，lca(u,v)和v，lca(u,v)（v，lca(u,v)是倒过来的）。这样就只要考虑自下而上的线性结构了，可以用前缀和的思想来做成段更新。

代码：

#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int maxn = 1e5 + ;
const int DEG = ;
const int mod = 1e9 + ;
typedef __int64 LL;

struct Edge {
    int v, type;
    Edge() {}
    Edge(int v, int type) :v(v), type(type) {}
};

int n,q;
vector<Edge> egs;
vector<int> G[maxn];

int dep[maxn], anc[maxn][DEG],up[maxn];
void dfs(int u,int fa) {
    dep[u] = dep[fa] + ;
    anc[u][] = fa;
    for (int i = ; i < DEG; i++) {
        int f = anc[u][i - ];
        anc[u][i] = anc[f][i - ];
    }
    for (int i = ; i < G[u].size(); i++) {
        Edge& e = egs[G[u][i]];
        if (e.v == fa) continue;
        up[e.v] = -e.type;
        dfs(e.v, u);
    }
}

int cnt[][maxn];
int Lca(int u, int v) {
    if (dep[u] < dep[v]) swap(u, v);
    for (int i = DEG - ; i >= ; i--) {
        if (dep[anc[u][i]]>=dep[v]) {
            u = anc[u][i];
        }
    }
    if (u == v) return u;
    for (int i = DEG - ; i >= ; i--) {
        if (anc[u][i] != anc[v][i]) {
            u = anc[u][i];
            v = anc[v][i];
        }
    }
    //printf("u:%d,v:%d\n", u, v);
    return anc[u][];
}

void dfs2(int u, int fa) {
    for (int i = ; i < G[u].size(); i++) {
        Edge& e = egs[G[u][i]];
        if (e.v == fa) continue;
        dfs2(e.v, u);
        cnt[][u] += cnt[][e.v];
        cnt[][u] += cnt[][e.v];
    }
}

void addEdge(int u, int v, int type) {
    egs.push_back(Edge(v, type));
    G[u].push_back(egs.size() - );
}

LL fast_pow(int n) {
    LL ret = ,x=;
    while (n) {
        if (n & ) ret *= x, ret %= mod;
        x *= x, x %= mod;
        n /= ;
    }
    return ret;
}

void init() {
    for (int i = ; i < maxn; i++) G[i].clear();
    egs.clear();
    memset(anc, , sizeof(anc));
    memset(dep, , sizeof(dep));
    memset(up, , sizeof(up));
    memset(cnt, , sizeof(cnt));
}

int main() {
    scanf("%d", &n);
    init();
    for (int i = ; i < n - ; i++) {
        int u, v, type;
        scanf("%d%d%d", &u, &v, &type);
        if (type == ) {
            addEdge(u, v, );
            addEdge(v, u, );
        }
        else {
            addEdge(u, v, );
            addEdge(v, u, -);
        }
    }
    dfs(, );
    //for (int i = 1; i <= n; i++) {
    //    printf("%d:(%d)",i,dep[i]);
    //    for (int j = 0; j < 20; j++) {
    //        printf("%d ", anc[i][j]);
    //    }
    //    printf("\n");
    //}
    scanf("%d", &q);
    int s = , t;
    while (q--) {
        //scanf("%d%d", &s, &t);
        //printf("(%d,%d):%d\n", s, t, Lca(s, t));
        scanf("%d", &t);
        int lca = Lca(s, t);
        //printf("(%d,%d):%d\n", s, t, lca);
        cnt[][s]++;
        cnt[][lca]--;
        cnt[][t]++;
        cnt[][lca]--;
        s = t;
    }
    dfs2(, );
    LL ans = ;
    for (int i = ; i <= n; i++) {
        if (up[i] == -) {
            ans += fast_pow(cnt[][i]) - ;
            ans = (ans + mod) % mod;
        }
        else if (up[i] == ) {
            ans += fast_pow(cnt[][i]) - ;
            ans = (ans + mod) % mod;
        }
    }
    printf("%I64d\n", ans);
    return ;
}
/*
7
1 2 0
1 3 0
2 4 0
2 5 0
3 6 0
3 7 0

*/