poj 1330 LCM 离线算法

http://poj.org/problem?id=1330 题目

 

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<vector>
using namespace std;

int n;
int fa[11111],ancestor[11111];
bool path[11111],indegree[11111];
int ans;
vector<int> son[11111];
int a,b;
void init(){for(int i=0;i<n;i++)fa[i]=i;}
int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
void merge(int x,int y)
{
    int xx=find(x),yy=find(y);
    if(xx!=yy)fa[yy]=xx;
}
void LCA(int x)
{
    path[x]=true,ancestor[x]=x;//分割当前子树，即当前子树的祖先都只可能是x
    for(std::vector<int>::iterator ii=son[x].begin();ii!=son[x].end();ii++)
    {
        int i=*ii;
        LCA(i);//处理孩子节点的子树
        merge(x,i);//将孩子节点并入x的树中
        ancestor[find(x)]=x;//将将i节点的子树的祖先都更新为x
    }
    /****
    查询a和b的公共祖先时有两种情况
        1.a是b的直接祖先，
            这时首先在b节点进入判断，ancestor为a
            再次进入a节点判断 ancestors还是a；
        2.a和b属于一个公共祖先的两个子树上，（设a相对b在左边子树）
            这时先在a中做下列判断，发现path[b]为false，
            再在b中做下列判断，这时ancestors[a]已经更新为当前树的根节点了。
    ****/
    if(x==a&&path[b])
        ans=ancestor[find(b)]+1;
    else if(x==b&&path[a])
        ans=ancestor[find(a)]+1;
}
int main()
{//freopen("in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        memset(indegree,false,sizeof(indegree));
        memset(path,false,sizeof(path));
        for(int i=0;i<n;i++)son[i].clear();
        init();
        for(int i=1;i<n;i++)
            scanf("%d%d",&a,&b),a--,b--,son[a].push_back(b),indegree[b]=true;
        scanf("%d%d",&a,&b),a--,b--;
        for(int i=0;i<n;i++)
            if(!indegree[i]){LCA(i);break;}
        cout<<ans<<endl;
    }
    return 0;
}