python-PAT-1065 单身狗-list超时问题

输入格式：
输入第一行给出一个正整数 N（≤ 50 000），是已知夫妻/伴侣的对数；随后 N 行，每行给出一对夫妻/伴侣——为方便起见，每人对应一个 ID 号，为 5 位数字（从 00000 到 99999），ID 间以空格分隔；之后给出一个正整数 M（≤ 10 000），为参加派对的总人数；随后一行给出这 M 位客人的 ID，以空格分隔。题目保证无人重婚或脚踩两条船。

输出格式：
首先第一行输出落单客人的总人数；随后第二行按 ID 递增顺序列出落单的客人。ID 间用 1 个空格分隔，行的首尾不得有多余空格。

输入样例：
3
11111 22222
33333 44444
55555 66666
7
55555 44444 10000 88888 22222 11111 23333

输出样例：
5
10000 23333 44444 55555 88888

---

这道题我写了三版。
流程都是建字典，然后筛选list，加入单身狗或者删除cp。

第一版我是删除cp，后两个case超时。

第二版做了一点改进，加了一个flag列表，跳过已经判断过的cp，这样能通过最后一个case，倒数第二个超时。

n = int(input())
cp = {}
for i in range(n):
    temp = input().split()
    cp[temp[0]] = temp[1]
    cp[temp[1]] = temp[0]
m = int(input())
info = input().split()
flags = [False] * m
ans = []
for i in range(m):
    if flags[i]: continue
    if info[i] in cp and cp[info[i]] in info:
        flags[i] = True
        flags[info.index(cp[info[i]])] = True
        continue
    else:
        ans.append(info[i])
res = sorted(ans)
print(len(res))
if len(res) > 0:
    print(" ".join(res))

不超时版

这里我一开始写ans给的类型是list，然后后两个case都超时了。。改成set就对了，说明set的 in 操作比 list 更快。

n = int(input())
cp = {}
for i in range(n):
    temp = input().split()
    cp[temp[0]] = temp[1]
m = int(input())
info = input().split()
ans = set(info)
for k,v in cp.items():
    if k in ans and v in ans:
        ans.remove(k)
        ans.remove(v)
res = sorted(ans)
print(len(res))
if len(res) > 0:
    print(" ".join(res))