使用的并查集,输出与下载的答案一致,但是不知道为什么全部是Wrong Answer
parent = {}
def find(parent, num):
if parent[num] != 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] = find(parent, f)
elif inp[0] == "?":
print(inp[1:], find(parent, inp[1:]))
elif inp[0] == "$":
break