hiho 13 最近祖先问题

问题描述

在树结构中，找到两个节点的最近最先节点。

解决方法

问题规模较小时，比如100个节点，我们可以先将一个节点的所有祖先存储起来，然后另外一个节点在查找祖先的过程中查看是否有相同的祖先。

std::map的使用：

判断key是否存在用count；

#include <cstdio>
#include <string>
#include <iostream>
#include <cstring>
#include <unordered_map>
#include <vector>
using namespace std;
unordered_map<string, int> nameToid;
vector<string> names;
enum {maxn = 200+4};
int f[maxn];

int findFirst(int p1, int p2)
{
    int pp[maxn];
    memset(pp, 0, sizeof(pp));
    for (int i= p1; i!= -1; i= f[i])
        pp[i]= 1;
    int i;
    for (i= p2; i!= -1 && !pp[i]; i= f[i])
        ;
    return i;
}

int main()
{
    memset(f, -1, sizeof(f));
    int n;
    scanf("%d", &n);
    for (int i=0; i< n; i++)
    {
        string father, son;
        cin>>father>>son;

        if (!nameToid.count(father))
        {
            nameToid[father] = names.size();
            names.push_back(father);
        }
        if (!nameToid.count(son))
        {
            nameToid[son] = names.size();
            names.push_back(son);
        }
        f[nameToid[son]] = nameToid[father];
    }
    int m;
    scanf("%d", &m);
    for (int i=0; i< m; i++)
    {
        string p1, p2;
        cin>>p1>>p2;
        if (p1 == p2)// 注意 如果名字相等则直接输出，名字不在树中也可以。
        {
            cout<<p1<<endl;
            continue;
        }
        int sp = -1;
        if (nameToid.count(p1) && nameToid.count(p2))
        {
            sp = findFirst(nameToid[p1], nameToid[p2]);
        }

        if (sp== -1)
            printf("-1\n");
        else
            cout<<names[sp]<<endl;

    }
    return 0;
}