2018年世界杯,冰岛队因1:1平了强大的阿根廷队而一战成名。好事者发现冰岛人的名字后面似乎都有个“松”(son),于是有网友科普如下:
冰岛人沿用的是维京人古老的父系姓制,孩子的姓等于父亲的名加后缀,如果是儿子就加 sson,女儿则加 sdottir。因为冰岛人口较少,为避免近亲繁衍,本地人交往前先用个 App 查一下两人祖宗若干代有无联系。本题就请你实现这个 App 的功能。
输入格式:
输入首先在第一行给出一个正整数 N(1<N≤105),为当地人口数。随后 N 行,每行给出一个人名,格式为:名 姓(带性别后缀)
,两个字符串均由不超过 20 个小写的英文字母组成。维京人后裔是可以通过姓的后缀判断其性别的,其他人则是在姓的后面加 m
表示男性、f
表示女性。题目保证给出的每个维京家族的起源人都是男性。
随后一行给出正整数 M,为查询数量。随后 M 行,每行给出一对人名,格式为:名1 姓1 名2 姓2
。注意:这里的姓
是不带后缀的。四个字符串均由不超过 20 个小写的英文字母组成。
题目保证不存在两个人是同名的。
输出格式:
对每一个查询,根据结果在一行内显示以下信息:
- 若两人为异性,且五代以内无公共祖先,则输出
Yes
; - 若两人为异性,但五代以内(不包括第五代)有公共祖先,则输出
No
; - 若两人为同性,则输出
Whatever
; - 若有一人不在名单内,则输出
NA
。
所谓“五代以内无公共祖先”是指两人的公共祖先(如果存在的话)必须比任何一方的曾祖父辈分高。
输入样例:
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
提交结果:
代码(测试点6非零返回):
import sys
def dfs(st):
vis[st] = 1
for i in range(len(ma[st])):
vis[ma[st][i]] = 1
dfs(ma[st][i])
def dfs2(st, k):
global judge
if k >= 3:
return
for i in range(len(ma[st])):
if vis[ma[st][i]]:
judge = False
return
vis[ma[st][i]] = 1
dfs2(ma[st][i], k + 1)
n = int(sys.stdin.readline())
re = [['', '', -1] for i in range(n + 1)] # 名 姓 性别
mp = {}
ma = [[] for _ in range(n + 1)]
for i in range(1, n + 1):
data = sys.stdin.readline().split()
re[i][0] = data[0]
re[i][1] = data[1]
mp[data[0]] = i
for i in range(1, n + 1):
k = re[i][1]
if k[-1] == 'm' or k[-1] == 'f':
if k[-1] == 'm':
re[i][2] = 1
else:
re[i][2] = 0
continue
else:
if k[-4:] == "sson":
re[i][2] = 1
ma[mp[re[i][0]]].append(mp[k[:-4]])
elif k[-7:] == "sdottir":
re[i][2] = 0
ma[mp[re[i][0]]].append(mp[k[:-7]])
k = int(sys.stdin.readline())
for i in range(k):
n1, s1, n2, s2 = sys.stdin.readline().split()
if n1 not in mp.keys() or n2 not in mp.keys():
print("NA")
elif re[mp[n1]][2] == re[mp[n2]][2]:
print("Whatever")
else:
vis = [0 for i in range(n + 1)]
x = 1
judge = True
dfs(mp[n1])
if vis[mp[n2]]:
x = 0
dfs2(mp[n2], 0)
x1 = x and judge
vis = [0 for i in range(n + 1)]
x = 1
judge = True
dfs(mp[n2])
if vis[mp[n1]]:
x = 0
dfs2(mp[n1], 0)
x2 = x and judge
if x1 and x2:
print("Yes")
else:
print("No")