HDU2586.How far away ？——最近公共祖先（离线Tarjan）

来源：互联网发布：矩阵变换编辑：程序博客网时间：2024/05/16 15:37

http://acm.hdu.edu.cn/showproblem.php?pid=2586

给定一棵带权有根树，对于m个查询(u,v)，求得u到v之间的最短距离

那么只要求得LCA(u,v)，dis（u,v)=dis[u]+dis[v]-2*dis[LCA(u,v)]，其中dis[i]表示节点i到根节点root的距离

31MS 4104K 2186 B

#include<iostream>#include<cstring>#include<cstdio>#include<algorithm>#include<string>#include<vector>const int maxn=40010;const int maxq=210;using namespace std;int n,m;struct Edge{    int to,w;    int next;}edge[maxn<<1];int head[maxn],tot;struct Query{    int to,next;    int index;}que[maxq<<1];int h[maxn],tt;void addedge(int u,int v,int w){    edge[tot].to=v;    edge[tot].w=w;    edge[tot].next=head[u];    head[u]=tot++;}void add_query(int u,int v,int index){    que[tt].to=v;    que[tt].index=index;    que[tt].next=h[u];    h[u]=tt++;}int f[maxn],dis[maxn],answer[maxq];bool vis[maxn],in[maxn];int find(int x){    return f[x]==-1 ? x:f[x]=find(f[x]);}void Union(int a,int b){    int t1=find(a);    int t2=find(b);    if(t1!=t2) f[t2]=t1;}void LCA(int u){    vis[u]=1;    for(int i=head[u];i!=-1;i=edge[i].next){        int v=edge[i].to;        if(vis[v]) continue;        dis[v]=dis[u]+edge[i].w;        LCA(v);        Union(u,v);    }    for(int i=h[u];i!=-1;i=que[i].next){        int v=que[i].to;        if(vis[v]){            answer[que[i].index]=dis[u]+dis[v]-2*dis[find(v)];        }    }}void Init(){    tot=tt=0;    memset(h,-1,sizeof(h));    memset(head,-1,sizeof(head));    memset(f,-1,sizeof(f));    memset(vis,false,sizeof(vis));    memset(in,false,sizeof(in));    memset(dis,0,sizeof(dis));}int main(){//#ifndef ONLINE_JUDGE//freopen("in.cpp","r",stdin);//#endif // ONLINE_JUDGE    int T,u,v,w;    cin>>T;    while(T--){        Init();        scanf("%d%d",&n,&m);        for(int i=0;i<n-1;++i){            scanf("%d%d%d",&u,&v,&w);            addedge(u,v,w);            addedge(v,u,w);            in[v]=true;        }        for(int i=0;i<m;++i){            scanf("%d%d",&u,&v);            add_query(u,v,i);            add_query(v,u,i);        }        for(int i=1;i<=n;++i){            if(in[i]==false){                LCA(i);                break;            }        }        for(int i=0;i<m;++i){            printf("%d\n",answer[i]);        }    }    return 0;}

0 0