【Lca 離線Tarjan演算法】hdu 2586 How far away ？

LCA 離線Tarjan演算法

題意： 對於一個圖，n(<=40000)個點，給出n-1條邊(u,v,w,)， m(<=200)個詢問給出兩點，問其最短距離。

Tarjan 離線求LCA：
離線演算法（必然先存所有詢問）；
dfs：
1. 遍歷到當前點，創造一個當前點的集合
2. 對於遍歷完的子樹，將其根節點的集合加入其父節點所在的集合。
3. 對於詢問的兩點，遍歷到一點，如果另一點已經被遍歷，那麼另一點已經加入集合的最高那一點即為最近公共祖先。
4. 因為另一點可能在此點的子樹上，所以詢問的答案應該在遍歷完其子樹後執行。

注意： 問題的記錄最好用前向星寫吧。

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
//#pragma comment(linker, "/STACK:102400000,102400000")

const double PI = acos(-1.0);
const double eps = 1e-6;
const int INF=0x3f3f3f3f;
const LL mod = 1e9+7;
const int N = 100000+10;
const int M = 2500000;

struct Node
{
    int from;
    int to;
    int val;
    int next;
};
int dir[N],fa[N],vis[N];
vector<Node> ma[N];     //vector 記錄圖
Node query[N*2];
int head[N];
int ans[N];
void addquery(int u,int v,int &top)
{
    query[++top].from = u; query[top].to = v; query[top].next = head[u];
    head[u] = top;
    query[++top].from = v; query[top].to = u; query[top].next = head[v];
    head[v] = top;
}
int Find(int n)
{
    if(fa[n] != n){
        fa[n] = Find(fa[n]);
    }
    return fa[n];
}
void dfs(int n,int dist)  //dfs處理需要陣列
{
    fa[n] = n;
    vis[n]=true;
    dir[n]=dist;
    for(int i = 0; i < ma[n].size(); i++){
        Node temp = ma[n][i];
        if(!vis[temp.to]){
            dfs(temp.to,dist+temp.val);
            fa[temp.to] = n;
        }
    }
    for(int i = head[n]; i != -1; i=query[i].next){
        int v = query[i].to;
        if(vis[v]){
            int pa = Find(v);
            //query[i].val = dir[n]+dir[v]-2*dir[pa];
            ans[(i-1)/2] = dir[n]+dir[v]-2*dir[pa];
        }
    }
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        memset(vis,0,sizeof(vis));
        memset(head,-1,sizeof(head));
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i = 1; i <= n; i++){
            ma[i].clear();
        }
        for(int i = 1; i < n; i++){
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            Node temp;
            temp.to = v; temp.val = w;
            ma[u].push_back(temp);
            temp.to = u; temp.val = w;
            ma[v].push_back(temp);
        }
        int top = 0;
        for(int i = 0; i < m; i++){
            int u,v;
            scanf("%d%d",&u,&v);
            addquery(u,v,top);
        }
        dfs(1,0);
        for(int i = 0; i < m; i++){
            printf("%d\n",ans[i]);
        }
    }
    return 0;
}