poj1330 Nearest Common Ancestors

题目链接

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

题意

给定一棵树和树上的两个结点，输出这两个结点的最近公共祖先。

思路

题目给定了两个结点a，b，我的思路是分别求出a，b的所有祖先，保存在va，vb中，然后遍历va和vb，由于va和vb中的结点是按离树根越来越近排列的，所以va和vb中第一个重合的元素即为a，b的最近公共祖先。在树的表示方面，使用数组来存储树。

代码

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;

const int N = 10000 + 10;
int p[N];           //p[i]表示结点i的双亲结点
vector<int> va;     //结点a的所有祖先结点
vector<int> vb;     //结点b的所有祖先结点

void find_ancestor(int c, vector<int>& v)   //找到c的祖先，存储在v中
{
    if(p[c] == 0)   //树的根结点
    {
        v.push_back(c);
        return;
    }

    v.push_back(c);
    find_ancestor(p[c], v);
}

void get_nca(int a, int b)
{
    va.clear();
    vb.clear();
    find_ancestor(a, va);
    find_ancestor(b, vb);

    int la = va.size();
    int lb = vb.size();

    for(int i=0; i<la; i++)
    {
        for(int j=0; j<lb; j++)
        {
            if(va[i]==vb[j])
            {
                cout<<va[i]<<endl;
                return;
            }
        }
    }
}

int main()
{
    //freopen("poj1330.txt", "r", stdin);
    int t;
    cin>>t;
    while(t--)
    {
        memset(p, 0, sizeof(p));
        int n;
        cin>>n;
        for(int i=0; i<n-1; i++)
        {
            int a, b;
            cin>>a>>b;
            p[b] = a;
        }
        int a, b;
        cin>>a>>b;
        get_nca(a, b);
    }
    return 0;
}