【HDU 2586】How far away?(最近公共祖先)

传送门：http://acm.hdu.edu.cn/showproblem.php?pid=2586

题意，求一棵树上任意两点的权值和。

分析：在没学最近公共祖先之前一直以为这是道最短路的题，学了最近公共祖先发现利用最近公共祖先去求解十分方便。拿到这个题最先的思路是用Tarjan求LCA的时候，当找到了LCA，用dis数组记录两个询问的点到LCA的距离，但是敲的过程中发现这样不好进行操作，最后发现u，v两点的距离=dis[u]+dis[v]-2*LCA[u,v]。而这样的话我们只需要在dfs的同时用dis数组记录这个点到根的距离即可。

本题还有一点需要注意的就是，因为是离线算法，输出应该按照输入的顺序输出，最开始我只是按照dfs顺序输出，wa了一发，才意识到要按顺序输出。

代码：

/* ***********************************************
Author        :Torrance_ZHANG
Created Time  :2016/5/1 12:12:02
File Name     :ceshi.cpp
************************************************ */

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
typedef long long int LL;
using namespace std;
const int MAXN=40010;
struct Node{
    int to;
    int next;
    int w;
    Node(){
        to=next=-1;
        w=0;
    }
}edge[MAXN];
int cnt=0,n,m,head[MAXN],fa[MAXN],dis[MAXN],deg[MAXN];
int vis[MAXN],ancient[MAXN],x[MAXN],y[MAXN];
void add(int u,int v,int w){
    edge[cnt].w=w;
    edge[cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt;
    cnt++;
}
int Find(int x){
    return x==fa[x]?fa[x]:fa[x]=Find(fa[x]);
}
void init(){
    memset(x,0,sizeof(x));
    memset(y,0,sizeof(y));
    memset(ancient,0,sizeof(ancient));
    memset(deg,0,sizeof(deg));
    memset(dis,0,sizeof(dis));
    memset(vis,0,sizeof(vis));
    memset(head,-1,sizeof(head));
}
void Tarjan(int u){
    vis[u]=1;
    fa[u]=u;
    for(int i=1;i<=m;i++){
        if(x[i]==u&&vis[y[i]]) ancient[i]=Find(y[i]);
        if(y[i]==u&&vis[x[i]]) ancient[i]=Find(x[i]);
    }
    for(int i=head[u];i!=-1;i=edge[i].next){
        if(!vis[edge[i].to]){
            dis[edge[i].to]=dis[u]+edge[i].w;
            Tarjan(edge[i].to);
            fa[Find(u)]=u;
        }
    }
}
int main(){
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        init();
        for(int i=1;i<n;i++){
            int x,y,w;
            scanf("%d%d%d",&x,&y,&w);
            add(x,y,w);
            deg[y]++;
        }
        int root=0;
        for(int i=1;i<=n;i++){
            if(deg[i]==0){
                root=i;
                break;
            }
        }
        for(int i=1;i<=m;i++){
            scanf("%d%d",&x[i],&y[i]);
        }
        Tarjan(root);
        for(int i=1;i<=m;i++)  
            printf("%d\n",dis[x[i]]+dis[y[i]]-2*dis[ancient[i]]);
    /*    for(int i=1;i<=n;i++)
            cout<<dis[i]<<endl;
        for(int i=1;i<=m;i++)
            cout<<ancient[i]<<endl;*/
    }
    return 0;
}


/*
3
3 2
1 2 10
3 1 15
1 2
2 3

2 2
1 2 100
1 2
2 1

3 2
1 2 10
3 1 15
2 3
1 2

*/

 

转载于:https://www.cnblogs.com/Torrance/p/5450545.html