POJ1330 最近公共祖先

寻找最近公共祖先可以说是树结构里相当基础的知识了。最近学了vector，正好可以派上用场。

本题采用双重链表并行上找的方式。即对于两个节点u和v，若u的层次比v深，则u去寻找父亲，否则v去寻找父亲，直至u=v。

程序中的fa[]数组用来记录父节点，step[]用以记录层次，a[]用以记录某一节点的儿子。函数中的findroot()用以寻找根节点，build()用于建立树结构，注意使用迭代器时需加上“vector :: iterator i”。

代码如下：

#include<iostream>
#include<vector>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=10005;
int t,n,fa[maxn],step[maxn];
vector<int> a[maxn];

int findroot()
{
    int root;
    for (int i=1;i<=n;++i)
    {
        if (fa[i]==0)
        {
            root=i;
            return root;
        }
    }
}

void build(int ste,int root)
{
    step[root]=ste; 
    vector <int> :: iterator i;
    for (i=a[root].begin();i != a[root].end();++i)
      build(ste+1,*i);
}

int find(int u,int v)
{
    int steu=step[u],stev=step[v];
    while (u!=v)
    {
        if (steu>stev)
        {
            steu--;
            u=fa[u];
        }
        else
        {
            stev--;
            v=fa[v];
        }       
    }
    return u;
}

void init()
{
    cin>>t;
    for (int i=1;i<=t;++i)
    {
        cin>>n;
        int u,v;
        memset(fa,0,sizeof(fa));
        memset(step,0,sizeof(step));
        for (int i=1;i<=n;++i)
          a[i].clear();
        for (int i=1;i<=n-1;++i)
        {
            cin>>u>>v;
            fa[v]=u;
            a[u].push_back(v);
        }
        build(0,findroot());
        cin>>u>>v;
        cout<<find(u,v)<<endl;
    }
}

int main()
{   
    init();
    return 0;       
}