7-8 生物进化 (10 分)

7-8 生物进化 (10 分)

在研究生物进化中，常用一种类似树状分支的图形来概括各种（类）生物之间的亲缘关系。下图（来源于网络）就是一棵生物进化树。

树可分为 有根树(rooted tree)和 无根树(unrooted tree)两类，为了便于讨论，这里我们只涉及有根树。有根树是具有方向的树，选择其中某个确定的节点，将其作为树中所有物种的共同祖先（根）。有根树这种结构在计算机科学中极为常见，尤其适用表述层次结构。下面就是在计算机科学中常见的有根树的示意图（注意：我们把树根放在了图的顶部，树叶放在了底部）。

给定2种生物的种类，请你判断，它们具有哪一个相同的祖先，当然由于树根一定满足这个要求，可能的答案不唯一，这里只要求你给出离这2种生物最近的祖先。 例如：对于上图，2，3的最近祖先是10，而9，7的最近祖先是8，特别需要注意，10，11的最近祖先是10。

输入格式

输入数据首先包含一个整数T(1<=T<=20)，表示测试实例的个数，然后是T组测试数据。 每组测试数据的第一行是1个整数n（节点(物种)从1到n编号，2<=n<=1000），表示节点总数，然后是n-1个整数对a, b（1<=a,b<=n，且a<>b），表示a是b的父节点(直接祖先)。请注意，n个节点(物种)的有根树恰好具有n-1条边。接下来是最多不超过1000次的询问（各占一行），每个询问包含2个互不相同的整数c，d（1<=c,d<=n），请你求解c,d的最近祖先。当cd0时，表示本组测试结束。

具体格式请参看Sample。

输出格式

对于每组测试数据, 分别按行输出求得的最近祖先。

输入样例

2
11
8 4
8 5
8 10
5 9
5 6
10 2
10 3
10 1
2 7
2 11
6 11
7 10
0 0
5
2 3
5 1
5 2
5 4
1 3
5 1
0 0

输出样例

8
10
5
5

基本思路

这题原本早就搞出来了，结果思维定势以为每组的测试样例就两个，一直被卡在那里，还是读题不仔细啊~
这题的思路跟“有情人”那题很像，dfs+标记法可以解决问题。

代码

#include <bits/stdc++.h>
using namespace std;
int t,n,root,comroot;//测试样例组数，结点个数，根结点，共同祖先
int v[1001];
bool exist[1001];//下标为物种编号，值为这个生物有没有父结点（用于求出根节点）
bool vis[1001];
void dfs(int x){
    if(vis[x]==1){
        comroot=x;
        return;
    }
    if(x==root){
        vis[x]=1;
        return;
    }
    vis[x]=1;
    dfs(v[x]);
}
int main(){
    cin>>t;
    while(t--){
        cin>>n;
        //构建关系图v
        for(int i=1;i<=n-1;i++){
            int ans1,ans2;
            cin>>ans1>>ans2;
            v[ans2]=ans1;
            exist[ans2]=1;
        }
        //找出根节点
        for(int i=1;i<=n;i++){
            if(exist[i]==false){
                root=i;
                break;
            }
        }
        //读入测试样例，求出共同祖先
        int ans1,ans2;
        while(cin>>ans1>>ans2){
            if(ans1==0||ans2==0){
                break;
            }else{
                fill(vis+1,vis+n+1,0);
                dfs(ans1);
                dfs(ans2);
                cout<<comroot<<endl;
            }
        }
        //初始化数据结构
        fill(exist+1,exist+n+1,0);
        fill(v+1,v+n+1,0);
    }
}