离线版本的思路是:每个点都把与它有关的查询放进它的那个vector里,然后对这棵树进行一次dfs,在dfs的过程中直接得出所有查询的答案,复杂度是O(n+q)。
具体来说,假设有一个查询u,v,当遍历到u的时候,如果v还没有访问,就先不管这个查询;如果v已经访问过了,
则由于进行的是深度优先搜索,当前结点与v的最近公共祖先一定还没有检查过,然而这个最近公共祖先的包含v的子树
一定已经搜索过了,那么这个最近公共祖先一定是v所在集合的祖先
#include <bits/stdc++.h>
using namespace std;
const int maxn=40005;
int f[maxn],head[maxn],dis[maxn],ans[202],vis[maxn];
int tot,n;
struct Edge{
int u,v,w,next;
};
Edge edge[maxn*2];
vector< pair<int,int> > Q[maxn];
int find(int x)
{
return f[x]==x?x:x=find(f[x]);
}
void add(int u,int v,int w)
{
edge[tot].u=u;
edge[tot].v=v;
edge[tot].w=w;
edge[tot].next=head[u];
head[u]=tot++;
}
void init()
{
for(int i=0;i<=maxn;i++)
Q[i].clear(),f[i]=i,ans[i]=0,dis[i]=0,vis[maxn],head[i]=-1;
tot=0;
}
void dfs(int u,int d)
{
dis[u]=d;
vis[u]=1;
for(int i=head[u];~i;i=edge[i].next)
{
int v=edge[i].v;
int w=edge[i].w;
if(!vis[v])
{
dfs(v,d+w);
f[v]=u;
}
}
for(int i=0;i<Q[u].size();i++)
{
int v=Q[u][i].first;
int id=Q[u][i].second;
if(vis[v])
{
int lca=find(v);
ans[id]=dis[u]-dis[lca]+dis[v]-dis[lca];
}
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
init();
int q;
scanf("%d %d",&n,&q);
for(int i=0;i<n-1;i++)
{
int u,v,w;
scanf("%d %d %d",&u,&v,&w);
add(u,v,w);
add(v,u,w);
}
for(int i=0;i<q;i++)
{
int u,v;
scanf("%d %d",&u,&v);
Q[u].push_back(make_pair(v,i));
Q[v].push_back(make_pair(u,i));//
}
dfs(1,0);
for(int i=0;i<q;i++)printf("%d\n",ans[i]);
}
return 0;
}