#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 40010, M = N + N;
int n, m;
int h[N], e[M], w[M], ne[M], idx;
int dist[N], fa[N][16], depth[N];
void add(int a, int b, int c)
{
e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++ ;
}
void dfs(int u, int father)
{
for (int i = h[u]; ~i; i = ne[i])
{
int j = e[i];
if (j == father) continue;
depth[j] = depth[u] + 1;
dist[j] = dist[u] + w[i];
fa[j][0] = u;
for (int k = 1; k <= 15; k ++ )
fa[j][k] = fa[fa[j][k - 1]][k - 1];
dfs(j, u);
}
}
int LCA(int a, int b)
{
if (depth[a] < depth[b]) return LCA(b, a);
for (int k = 15; k >= 0; k -- )
if (depth[fa[a][k]] >= depth[b])
a = fa[a][k];
if (a == b) return a;
for (int k = 15; k >= 0; k -- )
if (fa[a][k] != fa[b][k])
a = fa[a][k], b = fa[b][k];
return fa[a][0];
}
int main()
{
scanf("%d%d", &n, &m);
memset(h, -1, sizeof h);
for (int i = 0; i < n - 1; i ++ )
{
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
add(a, b, c), add(b, a, c);
}
//memset(depth, 0x3f, sizeof depth);
depth[1] = 1, depth[0] = 0;
dfs(1, -1);
while (m -- )
{
int a, b;
scanf("%d%d", &a, &b);
int p = LCA(a, b);
printf("%d\n", dist[a] + dist[b] - (dist[p] << 1));
}
return 0;
}
求关注
我不是早就关注了嘛
嗯。
当我看到你的求LCA的代码的时候,我想起很久之前我写的一份离线Tarjan求LCA的代码。闲来无事我也把自己的那个Tarjan粘过来跑了这个程序。发现比你的dfs更慢😹可是在线LCA一次求是logn的复杂度,m次那就是mlogn。离线是On+m的。哭晕
我也是跑了几次,LCA比离线快😄
我把yxc讲的Tarjan加上臭氧优化,优化到了180ms。我自己的代码有vector套vector,stl导致常数增大了。
LCA只有150😁
是的,树上倍增非常适合多次查询。