poj Nearest Common Ancestors（LCA+Tarjan）

【题目大意】：给定一棵树，求两点的最近公共祖先

【解题思路】：LCA问题，Tarjan离线算法可解。模版测试

【Tarjan算法】：

Tarjan算法是基于对图深度优先搜索的算法，每个强连通分量为搜索树中的一棵子树。搜索时，把当前搜索树中未处理的节点加入一个堆栈，回溯时可以判断栈顶到栈中的节点是否为一个强连通分量。

定义DFN(u)为节点u搜索的次序编号(时间戳)，Low(u)为u或u的子树能够追溯到的最早的栈中节点的次序号。由定义可以得出，

Low(u)=Min { DFN(u), Low(v),(u,v)为树枝边，u为v的父节点 DFN(v),(u,v)为指向栈中节点的后向边(非横叉边)}

当DFN(u)=Low(u)时，以u为根的搜索子树上所有节点是一个强连通分量。

算法伪代码如下

tarjan(u)
{
DFN[u]=Low[u]=++Index // 为节点u设定次序编号和Low初值
Stack.push(u)// 将节点u压入栈中
for each (u, v) in E // 枚举每一条边
if(v is not visted)// 如果节点v未被访问过
tarjan(v)// 继续向下找
Low[u]= min(Low[u], Low[v])
elseif(v in S)// 如果节点u还在栈内
Low[u]= min(Low[u], DFN[v])
if(DFN[u]== Low[u])// 如果节点u是强连通分量的根
repeat
v = S.pop// 将v退栈，为该强连通分量中一个顶点
print v
until (u== v)

【代码】：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#include <cmath>
#include <string>
#include <cctype>
#include <map>
#include <iomanip>
                   
using namespace std;
                   
#define eps 1e-8
#define pi acos(-1.0)
#define inf 1<<30
#define pb push_back
#define lc(x) (x << 1)
#define rc(x) (x << 1 | 1)
#define lowbit(x) (x & (-x))
#define ll long long
#define maxn 10005

int u,v;
int fa[maxn],rank[maxn],in[maxn],anc[maxn];
bool vist[maxn];
vector<int> vec[maxn],Q[maxn]; //vec保存边（父亲指向孩子u->v），Q保存询问（无向边,u->v && v->u）
int n, root; //找入度为0的点作为根做一次Tarjan就行了
void init() {
    for (int i=1; i<=n; i++) {
        rank[i]=1, fa[i]=i;
        in[i]=vist[i]=anc[i]=0;
        vec[i].clear(), Q[i].clear();
    }
}
int fr(int p){
    if (fa[p]!=p) fa[p]=fr(fa[p]);
    return fa[p];
}
bool un(int a,int b) {
    a=fr(a), b=fr(b);
    if (a == b) return 0;
    if (rank[a] <= rank[b])  fa[a]=b,rank[b]+=rank[a];
    else  fa[b]=a, rank[a]+=rank[b];
    return 1;
}
void Tarjan(int u){
    anc[u]=u;
    int size=vec[u].size();
    for (int i=0; i<size; i++){
        Tarjan(vec[u][i]);
        un(u,vec[u][i]);
        anc[fr(u)]=u;
    }
    vist[u]=1, size =Q[u].size();
    for (int i=0; i<size; i++){
        if (vist[Q[u][i]]==1){//如果已经访问了问题节点,就可以返回结果了.
            printf("%d\n",anc[fr(Q[u][i])]); //可以找到u，v的最近公共祖先
            return;
        }
    }
}

int main() {
    int T;
    cin >> T;
    while (T--){
        scanf("%d",&n);
        init();
        for(int i=0; i<n-1; i++) {
            scanf("%d%d",&u,&v);
            vec[u].pb(v);
            in[v]++;
        }
        scanf("%d%d",&u,&v);
        Q[u].pb(v),Q[v].pb(u);
        int root=-1;
        for(int i=1; i<=n; i++) {
            if(in[i]==0) {
                root=i;
                break;
            }
        }
        Tarjan(root);                
    }
    return 0;
}