PTA 家谱处理 (30 分)

家谱处理 (30 分)

人类学研究对于家族很感兴趣，于是研究人员搜集了一些家族的家谱进行研究。实验中，使用计算机处理家谱。为了实现这个目的，研究人员将家谱转换为文本文件。下面为家谱文本文件的实例：

家谱文本文件中，每一行包含一个人的名字。第一行中的名字是这个家族最早的祖先。家谱仅包含最早祖先的后代，而他们的丈夫或妻子不出现在家谱中。每个人的子女比父母多缩进2个空格。以上述家谱文本文件为例，John这个家族最早的祖先，他有两个子女Robert和Nancy，Robert有两个子女Frank和Andrew，Nancy只有一个子女David。

在实验中，研究人员还收集了家庭文件，并提取了家谱中有关两个人关系的陈述语句。下面为家谱中关系的陈述语句实例：

研究人员需要判断每个陈述语句是真还是假，请编写程序帮助研究人员判断。

输入格式:

输入首先给出2个正整数N（2≤N≤100）和M（≤100），其中N为家谱中名字的数量，M为家谱中陈述语句的数量，输入的每行不超过70个字符。

名字的字符串由不超过10个英文字母组成。在家谱中的第一行给出的名字前没有缩进空格。家谱中的其他名字至少缩进2个空格，即他们是家谱中最早祖先（第一行给出的名字）的后代，且如果家谱中一个名字前缩进k个空格，则下一行中名字至多缩进k+2个空格。

在一个家谱中同样的名字不会出现两次，且家谱中没有出现的名字不会出现在陈述语句中。每句陈述语句格式如下，其中X和Y为家谱中的不同名字：

X is a child of Y
X is the parent of Y
X is a sibling of Y
X is a descendant of Y
X is an ancestor of Y

输出格式:

对于测试用例中的每句陈述语句，在一行中输出True，如果陈述为真，或False，如果陈述为假。

输入样例:

6 5
John
  Robert
    Frank
    Andrew
  Nancy
    David
Robert is a child of John
Robert is an ancestor of Andrew
Robert is a sibling of Nancy
Nancy is the parent of Frank
John is a descendant of Andrew

输出样例:

True
True
True
False
False

#include<iostream>
#include<map>
using namespace std;
const int maxn=105;
struct N{string s;int su;};
N le[maxn];
int n,m,tot,p=1,fa[maxn];
string str;
map<string,int> mp;

int get(int x){
    if(fa[x]==0)    return x;
    return get(fa[x]);  //不能用压缩路径
}

void add(int a,int b){
    int x=get(a),y=get(b);
    if(x!=y)    fa[b]=a;
}

int main(){
    cin>>n>>m;
    (cin>>str).get();   //单独处理第一个人
    n--;    mp[str]=1;
    le[tot]={str,0};
    while(n--){
        getline(cin,str);
        int len=str.size(),cnt=0;
        while(str[cnt]==' ')    cnt++;  //计算缩进
        str.erase(0,cnt);   //删掉多余空格
        if(!mp.count(str))  mp[str]=++p;    //离散化
        while(!(cnt>le[tot-1].su))   tot--;
        if(cnt>le[tot].su) tot++;   //找到父结点
        add(mp[le[tot-1].s],mp[str]);
        le[tot].su=cnt; 
        le[tot].s=str;
    }
    string a,b,op;
    while(m--){
        cin>>a;cin>>op>>op>>op;
        cin>>b>>b;
        if(op=="child"){
            if(fa[mp[a]]==mp[b])
                cout<<"True\n";
            else    cout<<"False\n";
        }else if(op=="ancestor"){
            int x=mp[b],flag=0;
            while(x){
                if(x==mp[a])    {flag=1;break;}
                else    x=fa[x];
            }
            if(flag)    cout<<"True\n";
            else    cout<<"False\n";
        }else if(op=="sibling"){
            if(fa[mp[a]]==fa[mp[b]])    cout<<"True\n";
            else    cout<<"False\n";
        }else if(op=="descendant"){
            int x=mp[a],flag=0;
            while(x){
                if(x==mp[b])    {flag=1;break;}
                else    x=fa[x];
            }
            if(flag)    cout<<"True\n";
            else    cout<<"False\n";
        }else if(op=="parent"){
            if(fa[mp[b]]==mp[a]) cout<<"True\n";
            else    cout<<"False\n";
        }
    }
}