HDU 2586 How far way?

传送门

继续水板子题...

#include <bits/stdc++.h>
using namespace std;

inline int read() {
    int x = , f = ; char ch = getchar();
    while (ch < '' || ch > '') { if (ch == '-') f = -; ch = getchar(); }
    while (ch >= '' && ch <= '') { x = x *  + ch - ''; ch = getchar(); }
    return x * f;
}

const int maxn = 5e5 + ;

struct Edge { int to, next, c; } edge[ * maxn];
int cnt, head[maxn];
struct Qedge { int to, next; } qedge[ * maxn];
int qcnt, qhead[maxn], n, m;
int lca[maxn], dep[maxn], par[maxn];
bool vis[maxn];
inline void addedge(int u, int v, int c) {
    edge[++cnt].to = v;
    edge[cnt].c = c;
    edge[cnt].next = head[u];
    head[u] = cnt;
}
inline void addqedge(int u, int v) {
    qedge[++qcnt].to = v;
    qedge[qcnt].next = qhead[u];
    qhead[u] = qcnt;
}
int getfa(int x) { return x == par[x] ? x : par[x] = getfa(par[x]); }

void dfs(int u) {
    par[u] = u;
    vis[u] = ;
    for (int i = head[u]; i; i = edge[i].next) {
        int v = edge[i].to;
        if (!vis[v]) {
            dep[v] = dep[u] + edge[i].c;
            dfs(v);
            par[v] = u;
        }
    }
    for (int i = qhead[u]; i; i = qedge[i].next) {
        int v = qedge[i].to;
        if (vis[v]) {
            lca[i] = getfa(v);
            if (i % ) lca[i + ] = lca[i];
            else lca[i-] = lca[i];
        }
    }
}

int main() {
    int T = read();
    while (T--) {
        n = read();m = read();
        memset(head, , sizeof(head));
        memset(qhead, , sizeof(qhead));
        memset(vis, , sizeof(vis));
        memset(lca, , sizeof(lca));
        memset(dep, , sizeof(dep));
        cnt = qcnt = ;
        for (int i = ; i < n - ; i++)
        {
            int u = read(), v = read(), c = read();
            addedge(u, v, c);
            addedge(v, u, c);
        }
        for (int i = ; i < m; i++)
        {
            int u = read(), v = read();
            addqedge(u, v);
            addqedge(v, u);
        }
        dfs();
        for (int i = ; i <= m; i++)
        {
            int ans = dep[qedge[ * i].to] -  * dep[lca[ * i]] + dep[qedge[ * i - ].to];
            printf("%d\n", ans);
        }
    }
    return ;
}