题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2586
第一次用tarjan算法,留个模板。
#include<stdio.h>
#include<string.h>
#define spot 40010
#define edge 80010
#define query 410
struct stu
{
int to,value,next;
} a[edge];
int head[spot],x[query],y[query],z[query],sizes,dis[spot],fa[spot],n,m; //fa祖节点
bool vis[spot];
void init()
{
sizes=0;
memset(head,-1,sizeof(head));
memset(vis,0,sizeof(vis));
memset(dis,0,sizeof(dis));
for(int i=1; i<=n; i++)
fa[i]=i;
}
void add_edge(int from,int to,int value)
{
a[sizes].to=to;
a[sizes].value=value;
a[sizes].next=head[from];
head[from]=sizes++;
}
int finds(int x)
{
while(x!=fa[x])
return fa[x]=finds(fa[x]);
return x;
}
void tarjan(int u)
{
int k,v,i;
vis[u]=1;
for(i=1; i<=m; i++)
{
if(u==x[i]&&vis[y[i]]) z[i]=fa[y[i]];
if(u==y[i]&&vis[x[i]]) z[i]=fa[x[i]];
}
for(k=head[u]; k+1; k=a[k].next)
{
v=a[k].to;
if(!vis[v])
{
dis[v]=dis[u]+a[k].value;
tarjan(v);
fa[v]=u;
}
}
}
int main()
{
int nn;
scanf("%d",&nn);
while(nn--)
{
int from,to,value,i;
scanf("%d%d",&n,&m);
init();
for(i=1; i<n; i++)
scanf("%d%d%d",&from,&to,&value),add_edge(from,to,value),add_edge(to,from,value);
for(i=1; i<=m; i++)
scanf("%d%d",&x[i],&y[i]);
tarjan(1);
for(i=1; i<=m; i++)
printf("%d\n",dis[x[i]]+dis[y[i]]-2*dis[z[i]]);
}
return 0;
}