HDU2586 How far away ？（tarjan+并查集离线求lca）

传送门：http://acm.hdu.edu.cn/showproblem.php?pid=2586
这题大概是个模板题，又学了一个新姿势。先dfs一遍，求出每个点到根节点的距离。然后将询问离线，开始求lca，在dfs的时候同时更新并查集，就能求出当前位置下的已访问过的两个点的lca，两个点(a,b)的距离就是d[a]+d[b]-2*d[lca(a,b)];

#include <cstdio>#include <cstring>#include <algorithm>#include <cmath>#include <cstdlib>#include <cctype>#include <string>#include <iostream>#include <vector>#include <map>#include <set>#include <queue>#include <ctime>using namespace std;typedef long long ll;typedef pair<int,int> pii;#define pb push_back#define mp make_pair#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1#define calm (l+r)>>1const int INF = 1e9+7;const int maxn=40010;int n,m,Ecnt,Ecnt2,head[maxn],head2[maxn];struct EE{    int to,v,next;    EE(){}    EE(int t,int v,int n):to(t),v(v),next(n){}}edge[maxn<<1];struct E{    int to,next,lca;    E(){}    E(int t,int n):to(t),next(n){}}edge2[210];int dis[maxn],pre[maxn];ll ans[maxn];bool vis[maxn];inline void add(int a,int b,int c){    edge[Ecnt]=EE(b,c,head[a]);    head[a]=Ecnt++;}inline void add2(int a,int b){    edge2[Ecnt2]=E(b,head2[a]);    head2[a]=Ecnt2++;}void dfs(int s=1,int pre=0,int d=0){    dis[s]=d;    for(int i=head[s];~i;i=edge[i].next){        int t=edge[i].to;        if(t==pre)continue;        dfs(t,s,d+edge[i].v);    }}int find(int x){    if(pre[x]==x)return x;    return pre[x]=find(pre[x]);}void lca(int s=1){    pre[s]=s;    vis[s]=true;    for(int i=head[s];~i;i=edge[i].next){        int t=edge[i].to;        if(!vis[t]){            lca(t);            pre[t]=s;        }    }    for(int i=head2[s];~i;i=edge2[i].next){        if(vis[edge2[i].to]){            edge2[i].lca=find(edge2[i].to);            ans[i]=dis[s]+dis[edge2[i].to]-2*dis[edge2[i].lca];        }    }}int main(){    //freopen("D://input.txt","r",stdin);    int T;scanf("%d",&T);    while(T--){        scanf("%d%d",&n,&m);        memset(head,-1,sizeof head);        memset(head2,-1,sizeof head2);        Ecnt=Ecnt2=0;        for(int i=1;i<n;i++){            int a,b,c;scanf("%d%d%d",&a,&b,&c);            add(a,b,c);add(b,a,c);        }        for(int i=0;i<m;i++){            int a,b;scanf("%d%d",&a,&b);            add2(a,b);        }        memset(dis,0,sizeof dis);        memset(vis,false,sizeof vis);        dfs();        lca();        for(int i=0;i<m;i++){            printf("%I64d\n",ans[i]);        }    }    return 0;}