PTA:(天梯赛)冰岛人(25分)(AC满分+解析)

L2-2 冰岛人 (25分)

2018年世界杯,冰岛队因1:1平了强大的阿根廷队而一战成名。好事者发现冰岛人的名字后面似乎都有个“松”(son),于是有网友科普如下:
在这里插入图片描述
在这里插入图片描述
输入样例:

15
chris smithm
adam smithm
bob adamsson
jack chrissson
bill chrissson
mike jacksson
steve billsson
tim mikesson
april mikesdottir
eric stevesson
tracy timsdottir
james ericsson
patrick jacksson
robin patricksson
will robinsson
6
tracy tim james eric
will robin tracy tim
april mike steve bill
bob adam eric steve
tracy tim tracy tim
x man april mikes

输出样例:

Yes
No
No
Whatever
Whatever
NA

思路
1. 由题目知,每个人的“名”是不相同的,所有可以以“名”为标识符,使用map对每个人按照输入顺序编号。
2. 将维京人姓后面的后缀去掉(去掉后缀的姓,就是其父亲的名)
3. 此时,每个人的名对应编号我们就知道了,再根据每个人后面的姓,就可以知道其父亲 名的编号。(此处也可以使用并查集,因为每个人最多就一个长辈,我这里是直接不停循环遍历每个人的长辈)
4. 循环遍历待检验人的父母,判断是否相同,相同的话在五代以内。(这里要注意一下,相同的话,必须要在必须比任何一方的曾祖父辈分高。;换句话说,例:共同祖先是A的爸爸,是B的十几代祖宗,这样还是不可以的,因为在A的五代以内;或者B直接是A的直系亲属也有可能)
具体解释见AC代码:

#include<bits/stdc++.h>
#define inf 110000
using namespace std;
map<string, int> mp;//string:名   int:编号
int p[inf],cnt[inf],dx[inf],dy[inf],sex[inf]; 
string ming[inf], xing[inf];  //保存每个人的名和姓 

bool judge(int x, int y)
{
	memset(cnt,0,sizeof(cnt));
	memset(dx,0,sizeof(dx));
	memset(dy,0,sizeof(dy)); 
    cnt[x]=1,cnt[y]=1;
    while(p[x] != -1){  //如果x有长辈时,不停的向上循环,直到没有长辈为止 
        cnt[p[x]]++;
        dx[p[x]] = dx[x] + 1; //统计多少代了 
        if(p[x] == y) return false; //如果x的上辈等于y,直接返回,输出No 
        x = p[x];
    }
    while(p[y] != -1){
        cnt[p[y]]++;
        dy[p[y]] = dy[y] + 1;
        if(cnt[p[y]] > 1){ //>1说明p[y]也是x的长辈,判断多少代,是否达到五代 
            if(dy[p[y]]>=4 && dx[p[y]] >= 4) return true;
            else return false;
        }
        y = p[y];
    }
    return true;
}

int main(){
	int n,m;
	cin>>n;
	getchar();
	for(int i=1; i<=n; i++){
		cin>>ming[i]>>xing[i];
		mp[ming[i]]=i;  //名字是唯一的,所以将名改为编号,方便处理 
		int t=xing[i].size()-1;
		if(xing[i][t]=='m'||xing[i][t]=='n') sex[i]=1;//男(m字符结尾或sson结尾的名字) 
		else sex[i]=-1; //女	
		
		//保存将后缀去掉的姓,即父亲的名 
		string s="";
		if(xing[i][t]=='n'){
			for(int j=0; j<t-3; j++) s+=xing[i][j];
		}else if(xing[i][t]=='r'){
			for(int j=0; j<t-6; j++) s+=xing[i][j];
		}
		xing[i]=s;  
	}	
	memset(p,-1,sizeof(p));//初始为-1,表示无长辈
	for(int i=1; i<=n; i++){
		if(mp[xing[i]]>0) p[i]=mp[xing[i]];  //p[i]表示编号 i 它父亲编号 
	} 
	cin>>m;
	string s1,s2,s3,s4;
	while(m--){
		cin>>s1>>s2>>s3>>s4;
		int x=mp[s1],y=mp[s3];  
		if(!sex[x]||!sex[y]) cout<<"NA"<<endl;  //sex[i]为1是男,为-1是女,为0表示没有此人 
		else if(sex[x]==sex[y])cout<<"Whatever"<<endl;//先要判断是否存在,然后在判断是否同性,不然有测试点过不了  
		else{ //异性的话,判断是否为近亲 
			if(judge(x, y)) cout<<"Yes"<<endl;
			else cout<<"No"<<endl;			
		}			
	} 
	return 0;
} 

如果读者理解的还不是很透彻的话,参考下面图解:
在这里插入图片描述
图解参考链接:https://blog.csdn.net/weixin_43264529/article/details/88925108
欢迎大家批评改正!!!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章