hihocoder #1062 : 最近公共祖先·一(小数据量 map+set模拟+标记检查 *【模板】思路 )

#1062 : 最近公共祖先·一

时间限制: 10000ms
单点时限: 1000ms
内存限制: 256MB

描述

小Ho最近发现了一个神奇的网站!虽然还不够像58同城那样神奇,但这个网站仍然让小Ho乐在其中,但这是为什么呢?

“为什么呢?”小Hi如是问道,在他的观察中小Ho已经沉迷这个网站一周之久了,甚至连他心爱的树玩具都弃置一边。

“嘿嘿,小Hi,你快过来看!”小Ho招呼道。

“你看,在这个对话框里输入我的名字,在另一个对话框里,输入你的名字,再点这个查询按钮,就可以查出来……什么!我们居然有同一个祖祖祖祖祖爷爷?”

“诶,真是诶……这个网站有点厉害啊。”小Hi不由感叹道。

“是啊,这是什么算法啊,这么厉害!”小Ho也附和道。

“别2,我说的是他能弄到这些数据很厉害,而人类的繁殖树这种层数比较浅的树对这类算法的要求可是简单的不得了,你都能写出来呢!”小Hi道。

“啊?我也能写出来?可是……该从哪开始呢?”小Ho困惑了。

小Ho要面临的问题是这样的,假设现在他知道了N个人的信息——他们的父亲是谁,他需要对于小Hi的每一次提问——两个人的名字,告诉小Hi这两个人的是否存在同一个祖先,如果存在,那么他们的所有共同祖先中辈分最低的一个是谁?

提示:不着急,慢慢来,另外我有一个问题:挖掘机技术哪家强?!

输入

每个测试点(输入文件)有且仅有一组测试数据。

每组测试数据的第1行为一个整数N,意义如前文所述。

每组测试数据的第2~N+1行,每行分别描述一对父子关系,其中第i+1行为两个由大小写字母组成的字符串Father_i, Son_i,分别表示父亲的名字和儿子的名字。

每组测试数据的第N+2行为一个整数M,表示小Hi总共询问的次数。

每组测试数据的第N+3~N+M+2行,每行分别描述一个询问,其中第N+i+2行为两个由大小写字母组成的字符串Name1_i, Name2_i,分别表示小Hi询问中的两个名字。

对于100%的数据,满足N<=10^2,M<=10^2, 且数据中所有涉及的人物中不存在两个名字相同的人(即姓名唯一的确定了一个人)。

输出

对于每组测试数据,对于每个小Hi的询问,输出一行,表示查询的结果:如果根据已知信息,可以判定询问中的两个人存在共同的祖先,则输出他们的所有共同祖先中辈分最低的一个人的名字,否则输出-1。

样例输入
11

JiaYan JiaDaihua

JiaDaihua JiaFu

JiaDaihua JiaJing

JiaJing JiaZhen

JiaZhen JiaRong

JiaYuan JiaDaishan

JiaDaishan JiaShe

JiaDaishan JiaZheng

JiaShe JiaLian

JiaZheng JiaZhu

JiaZheng JiaBaoyu

3

JiaBaoyu JiaLian

JiaBaoyu JiaZheng

JiaBaoyu LinDaiyu
样例输出
JiaDaishan

JiaZheng

-1

代码:
#include <iostream>

#include <string>

#include <stdio.h>

#include <string.h>

#include <algorithm>

#include <map>

#include <set>



using namespace std;

map<string, string>ma;



int main()

{

    int n, m;

    int i, j;

    string s, t;

    string x, y;

    scanf("%d", &n);

    ma.clear();

    for(i=0; i<n; i++)

    {

        cin>>s>>t;

        ma[t]=s; //儿子指向父亲

    }

    set<string>q;

    scanf("%d", &m);

    for(j=0; j<m; j++)

    {

        q.clear(); //每次都必须要将set集合清空,否则WA

        cin>>x>>y;

        //对x进行操作

        string t=x;

        q.insert(t);

        while(ma.find(t)!=ma.end())//如果x存在“值”

        {

            q.insert(ma[t]); //将儿子对应的父亲插入到set的集合里面

            t=ma[t];

        }

        //对y进行操作

        int flag=0;

        if(q.find(y)!=q.end()) //说明y自身在x的祖先中存在吗

        {

            cout<<y<<endl;

            flag=1;

            continue;

        }

        while(ma.find(y)!=ma.end()) //将y往上找祖先

        {

            y=ma[y];

            if(q.find(y)!=q.end()) //说明已经找到

            {

                cout<<y<<endl;

                flag=1;

                break;

            }

        }

        if(flag==0)

            printf("-1\n");

    }

    return 0;

}

 

你可能感兴趣的:(code)