使用并查集写的,输出的答案和答案文件甚至换行符都一样
parent = {}
def find(parent, num):
if parent[num] == num:
return num
parent[num] = find(parent, parent[num])
return parent[num]
f = ""
while True:
inp = input()
if inp[0] == "#":
f = inp[1:]
if f not in parent.keys():
parent[f] = f
elif inp[0] == "+":
name = inp[1:]
parent[name] = f
elif inp[0] == "?":
print(inp[1:], find(parent, inp[1:]))
elif inp[0] == "$":
break