PAT~乙级~1065 单身狗 ~python-蒲公英云

PAT~乙级~1065 单身狗 ~python

题目描述：

“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人，以便给予特殊关爱。

输入格式：

输入第一行给出一个正整数 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

-————————————————————————————-答题区—————————————————————————————

num = int(input())
couple_lst = [-1]*100000
couple_lst_copy = [0]*100000
watch_lst = []
for i in range(num):
    id1,id2=map(int,input().split())
    couple_lst[id1] = id2
    couple_lst[id2] = id1
num = int(input())
join_lst = list(map(int,input().split()))
alone_lst = []
for i in join_lst:
    if(couple_lst[i]==-1):
        alone_lst.append(i)
    else:
        couple_lst_copy[i]=1
        watch_lst.append(i)
for i in watch_lst:
    if(couple_lst_copy[couple_lst[i]]!=1):
        alone_lst.append(i)   
k = len(alone_lst)
print(k)
alone_lst.sort()
if(k!=0):
    print('%05d'%alone_lst[0],end='')
    for i in alone_lst[1:]:
        print(' '+'%05d'%i,end='')