hdu 2856 最近公共祖先 离线

http://www.cnblogs.com/ylfdrib/archive/2010/11/03/1867901.html

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<utility>
#include<queue>
#include<set>
#include<map>
#include<math.h>
using namespace std;
#define inf 0x3f3f3f3f
#pragma warning(disable:4996)
#pragma comment(linker, "/STACK:102400000,102400000")
#define ll long long
const double eps=1e-9;

struct node
{
    int b,len;
};
vector<node> f[40040];
vector<node> h[40040];

int parent[40040];
int root(int x)
{
    if(parent[x]==-1) return x;
    else return parent[x]=root(parent[x]);
}
int vis[40040];
int dis[40040];
int ans[40040];
void tarjan(int x)
{
    vis[x]=1;

    for(int i=0;i<f[x].size();i++)
    {
        int v=f[x][i].b;
        if(vis[v]==-1)
        {
            dis[v]=dis[x]+f[x][i].len;
            tarjan(v);
            parent[v]=x;
        }
    }
    for(int i=0;i<h[x].size();i++)
    {
        int v=h[x][i].b;
        int id=h[x][i].len;
        if(vis[v]==1)
            ans[id]=dis[x]+dis[v]-2*dis[root(v)];
    }
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int n,m,i,j,k;
        cin>>n>>m;
        for(i=0;i<=n;i++)
        {
            f[i].clear();
            h[i].clear();
            parent[i]=-1;
            vis[i]=-1;
            dis[i]=0;
        }
        for(i=0;i<n-1;i++)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            node t;
            t.b=b,t.len=c;
            f[a].push_back(t);
            t.b=a;
            f[b].push_back(t);
        }
        for(i=0;i<m;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            node t;
            t.b=b,t.len=i;
            h[a].push_back(t);
            t.b=a;
            h[b].push_back(t);
        }
        tarjan(1);
        for(i=0;i<m;i++)
            printf("%d\n",ans[i]);
    }
}