HDU 2586 How far away ？【LCA】

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

How far away ？

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 24439    Accepted Submission(s): 9739

Problem Description

There are n houses in the village and some bidirectional roads connecting them. Every day peole always like to ask like this "How far is it if I want to go from house A to house B"? Usually it hard to answer. But luckily int this village the answer is always unique, since the roads are built in the way that there is a unique simple path("simple" means you can't visit a place twice) between every two houses. Yout task is to answer all these curious people.

 

Input

First line is a single integer T(T<=10), indicating the number of test cases.
  For each test case,in the first line there are two numbers n(2<=n<=40000) and m (1<=m<=200),the number of houses and the number of queries. The following n-1 lines each consisting three numbers i,j,k, separated bu a single space, meaning that there is a road connecting house i and house j,with length k(0<k<=40000).The houses are labeled from 1 to n.
  Next m lines each has distinct integers i and j, you areato answer the distance between house i and house j.

 

Output

For each test case,output m lines. Each line represents the answer of the query. Output a bland line after each test case.

 

Sample Input

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

 

Sample Output

10 25 100 100

题意概括：

给一颗 N 个结点，N-1条边的树，和 M 次查询（离线）

解题思路：

要查询两点到最近公共祖先的距离和，首先要预处理出根结点到各个结点的距离（DFS）

然后 Tarjan 找出最近的公共祖先，两个结点到根结点的距离之和减去两倍的最近公共祖先到根结点的距离就是该查询的答案了。

AC code：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#define INF 0x3f3f3f3f
#define LL long long
const int MAX_N = 4e4+5;
using namespace std;

struct Edge{int v, nxt, w;}edge[MAX_N<<1];              //静态邻接表
struct Query
{
    int v, id;
    Query(){};
    Query(int _v, int _id):v(_v), id(_id){};
};

vector<Query> q[MAX_N];       // !!!                    //查询关系动态二维矩阵

int head[MAX_N], cnt;                                   //静态邻接表头节点，边数
int dis[MAX_N];                                         //各结点到根结点的距离
int fa[MAX_N];                                          //并查集 祖先数组
bool vis[MAX_N], in[MAX_N];                             //深搜用，找根结点用
int ans[MAX_N];                // !!!                   //各查询答案
int N, M;                                               //结点数，查询数

void init()
{
    memset(head, -1, sizeof(head));                     //初始化
    memset(vis, false, sizeof(vis));
    memset(in, false, sizeof(in));
    memset(dis, 0, sizeof(dis));
    memset(ans, 0, sizeof(ans));
    cnt = 0;
}

void AddEdge(int from, int to, int weight)              //静态邻接表加边操作
{
    edge[cnt].v = to;
    edge[cnt].w = weight;
    edge[cnt].nxt = head[from];
    head[from] = cnt++;
}

int getfa(int x)                                        //找祖先
{
    int root = x;
    while(fa[root] != root) root = fa[root];

    int tmp;
    while(fa[x] != root){
        tmp = fa[x];
        fa[x] = root;
        x = tmp;
    }
    return root;
}

void dfs(int s)                                         //预处理各根结点到祖先的距离
{
    int rt = s;
    for(int i = head[s]; i != -1; i = edge[i].nxt){
        dis[edge[i].v] = dis[rt] + edge[i].w;
        dfs(edge[i].v);
    }
}

void Tarjan(int s)                                      // LCA
{
    int root = s;
    fa[s] = s;
    for(int i = head[s]; i != -1; i = edge[i].nxt){
        int Eiv = edge[i].v;
        Tarjan(Eiv);
        fa[getfa(Eiv)] = s;
    }
    vis[s] = true;
    for(int i = 0; i < q[s].size(); i++){
        if(vis[q[s][i].v] && !ans[q[s][i].id]){
            ans[q[s][i].id] = dis[q[s][i].v] + dis[s] - 2*dis[getfa(q[s][i].v)];
        }
    }
}

int main()
{
     int T_case;
     scanf("%d", &T_case);
     while(T_case--){
        init();
        scanf("%d %d", &N, &M);
        for(int i = 1, u, v, w; i < N; i++){                //建树
            scanf("%d %d %d", &u, &v, &w);
            AddEdge(u, v, w);
            in[v] = true;
        }

        for(int i = 1, u, v; i <= M; i++){                  //储存查询信息（离线）
            scanf("%d %d", &u, &v);
            q[u].push_back(Query(v, i));
            q[v].push_back(Query(u, i));
        }

        int root = 0;
        for(int i = 1; i <= N; i++) if(!in[i]){root = i; break;}    //找树根结点

        dfs(root);              //预处理
        Tarjan(root);           //LCA

        for(int i = 1; i <= M; i++){
            printf("%d
", ans[i]);
        }
        //puts("");
     }
     return 0;
}