题目:hdu 6110 路径交
分析:建好树之后dfs获得每个节点的深度,然后建立一颗线段树,每个节点维护一个路径,表示其左子节点维护的路径和右子节点维护的路径相交的路径(一棵树上两个节点相交只有一条路径),这里需要用到lca:即两个路径:a -> b, c -> d相交的路径为这四个节点(lca(a, c), lca(a, d), lca(b, c), lca(b, d))中深度较大的两个节点所连成的路径;
求两点a, b的距离可用公式:dis(a) + dis(b) - 2*dis(lca(a, b)), 其中,dis(a)代表a到达根节点的距离。这样可做到每次查询的时间为log(n)
代码如下:
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;typedef pair<int, int> P;
typedef long long ll;const int INF = (int)1e9;
const int maxn = 500000 + 10;int cnt, lst[maxn], nxt[maxn], to[maxn], c[maxn];
int fa[2*maxn][20], dep[2*maxn];
ll dis[2*maxn];P p[2*maxn];
int n, m;
int q1, q2;void add(int u, int v, int d){nxt[++cnt] = lst[u]; lst[u] = cnt; to[cnt] = v, c[cnt] = d;nxt[++cnt] = lst[v]; lst[v] = cnt; to[cnt] = u, c[cnt] = d;
}void dfs(int u, int f){for (int j = lst[u]; j; j = nxt[j]){int v = to[j];if (v == f) continue;dis[v] = dis[u] + c[j];dep[v] = dep[u] + 1;fa[v][0] = u;dfs(v, u);}}void init_lca(){for (int j = 1; (1 << j) <= n; j ++)for (int i = 1; i <= n; i ++)fa[i][j] = fa[fa[i][j-1]][j-1];
}int lca(int a, int b) {if (dep[a] < dep[b]) swap(a, b);int f = dep[a] - dep[b];for (int j = 0; (1 << j) <= f; j ++)if ((1 << j) & f) a = fa[a][j];if (a != b) {for (int j = log2(n) + 1; j >= 0; j --) {if (fa[a][j] != fa[b][j]){a = fa[a][j], b = fa[b][j];}}a = fa[a][0];}return a;}bool cmp(int a, int b) {return dep[a] > dep[b];
}P unit(P t1, P t2){int a[10];a[0] = lca(t1.first, t2.first);a[1] = lca(t1.first, t2.second);a[2] = lca(t1.second, t2.first);a[3] = lca(t1.second, t2.second);sort(a, a + 4, cmp);return P(a[0], a[1]);}void build(int o, int L, int R) {if (L == R) {scanf("%d %d", &p[o].first, &p[o].second);return;}int M = L + (R-L) / 2;build(2*o, L, M);build(2*o + 1, M + 1, R);p[o] = unit(p[2*o], p[2*o + 1]);
}P query(int o, int L, int R) {if (q1 <= L && q2 >= R) {return p[o];}int M = L + (R-L) / 2, lc = 2*o, rc = 2*o + 1;if (q2 <= M) return query(lc, L, M);else if (q1 > M) return query(rc, M + 1, R);else return unit(query(lc, L, M), query(rc, M+1, R));
}int main(){freopen("a.in", "r", stdin);freopen("a.out", "w", stdout);scanf("%d", &n);for (int i = 1; i < n; i ++) {int u, v, d;scanf("%d %d %d", &u, &v, &d);add(u, v, d);}dfs(1, 1);init_lca();scanf("%d", &m);build(1, 1, m);int kase; scanf("%d", &kase);while (kase --) {scanf("%d %d", &q1, &q2);P a = query(1, 1, m);ll ans = dis[a.first] + dis[a.second]-2*dis[lca(a.first, a.second)];printf("%lld\n", ans);}return 0;
}