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
歡迎大家批評改正!!!

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