欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【PTA】家谱处理 (20 分)——map映射

程序员文章站 2024-03-16 16:31:40
...

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

John
  Robert
    Frank
    Andrew
  Nancy
    David

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

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

John is the parent of Robert
Robert is a sibling of Nancy
David is a descendant of Robert

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

输入格式:
输入首先给出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

用map映射关系做,参考这里。

其中,祖先的和后代的要有循环的操作。
ps:如果用[]来访问vector的元素,要开好长度,否则就用push_back输入。
同级兄弟间l的cnt可以覆盖,在father的映射对应了就行。

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int n,m,cnt=0;
	cin>>n>>m;
	getchar();
	map<string,string>father;
	vector<string>l(110);
	string name;
	for(int i=0;i<n;i++)
	{		
		getline(cin,name);
		
		cnt=count(name.begin(),name.end(),' ');
		name=name.substr(cnt);
		if(cnt==0)//是祖先
		{
			l[0]=name;
			father[name]="root";
		} 
		else
		{
			father[name]=l[cnt/2-1];
			l[cnt/2]=name;//可以直接覆盖,因为它与上面的是兄弟,但下面的若是孩子,则是当前兄弟的孩子			 
		}
		
	}
	
	string a,b,c,d;
	while(m--)
	{
		cin>>a>>d>>d>>b>>d>>c;//a是c的b
		if(b=="child")
		{
			if(father[a]==c) cout<<"True"<<endl;
			else cout<<"False"<<endl;
		 } 
		
		else if(b=="ancestor")
		{
			while(father[c]!=a&&father[c]!="root") c=father[c];//往上走 
			if(father[c]=="root")cout<<"False"<<endl;//上级不是a 
			else cout<<"True"<<endl;
		}
		else if(b=="sibling")
		{
			if(father[a]==father[c])cout<<"True"<<endl;
			else cout<<"False"<<endl;
		}
		else if(b=="parent")
		{
			if(father[c]==a)cout<<"True"<<endl;
			else cout<<"False"<<endl;
		}
		else
		{
			while(father[a]!=c&&father[a]!="root") a=father[a];
			if(father[a]=="root")cout<<"False"<<endl;
			else cout<<"True"<<endl;
		}
	}
	return 0;
}