很裸的LCA问题!
Tarjan算法:
#include<iostream>
#include<vector>
#include<cstring>
#include<cstdlib>
using namespace std;
struct node
{
int t,cap;
};
vector<node> qry[40010];
vector<node> map[40010];
int ans[210];
int parent[40010];
int dist[40010];
int find(int x)
{
if(parent[x]!=x)
return parent[x]=find(parent[x]);
return x;
}
void LCA(int u)
{
parent[u]=u;
for(int i=0;i<qry[u].size();i++)
{
int v=qry[u][i].t;
if( parent[v] && ans[qry[u][i].cap]==-1 )
ans[qry[u][i].cap]=dist[v]+dist[u]-2*dist[find(v)];
}
for(int i=0;i<map[u].size();i++)
{
int v=map[u][i].t;
if( !parent[v] )
{
dist[v]+=(dist[u]+map[u][i].cap);
LCA(v);
parent[v]=u;
}
}
}
int main()
{
// freopen("in.txt","r",stdin);
int t;
scanf("%d",&t);
int n,m;
while(t--)
{
memset(ans,-1,sizeof(ans));
memset(parent,0,sizeof(parent));
memset(dist,0,sizeof(dist));
for(int i=0;i<40010;i++)
{
qry[i].clear();
map[i].clear();
}
scanf("%d %d",&n,&m);
for(int i=0;i<n-1;i++)
{
int a,b,k;
node x;
scanf("%d %d %d",&a,&b,&k);
x.t=b; x.cap=k;
map[a].push_back(x);
x.t=a; x.cap=k;
map[b].push_back(x);
}
for(int i=0;i<m;i++)
{
int a,b;
scanf("%d %d",&a,&b);
node x;
x.t=b; x.cap=i;
qry[a].push_back(x);
x.t=a; x.cap=i;
qry[b].push_back(x);
}
LCA(1);
for(int i=0;i<m;i++)
printf("%d\n",ans[i]);
}
return 0;
}