1065 单身狗 python

1065 单身狗 (25 分)

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

输入格式:

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

我的思路是建立字典,把情侣放进去,以输入样例为例,字典1是{11111:22222,33333:44444,55555:66666} ,然后把keys和values翻转,得到一个字典2(就不写出来了)

然后去遍历参加的客人id列表,如果id对应了字典1中key1的值,那么再去判断key1的值是否在客人id列表中,不在的话说明是一个单身狗。同理在对字典2做判断。如果不在字典1或者字典2里,那么直接判定为单身狗

N=int(input())

dic_lover={}
for i in range(N):
    ls=input().split()
    dic_lover[ls[0]]=ls[1]
dic_lover_re={}
for key in dic_lover:   #做了个翻转的字典
    dic_lover_re[dic_lover[key]]=key

M=int(input())
ls_id=input().split()

##print(dic_lover)
##print('******')
##print(dic_lover_re)
##print(ls_id)
ls_dog=[]

for i in ls_id:
    if i in dic_lover:
        if dic_lover[i] not in ls_id:
            ls_dog.append(i)
    elif i in dic_lover_re:
        if dic_lover_re[i] not in ls_id:
            ls_dog.append(i)
    else:
        ls_dog.append(i)
ls_dog.sort()
print(len(ls_dog))

if len(ls_dog) !=0:    #测试点1 是没有单身狗的情况,此时第二行应该没有输出
    print(' '.join(ls_dog))
#测试点4有时候会超时,有时候不会

 测试点3超时~~~

猜你喜欢

转载自blog.csdn.net/weixin_43731183/article/details/86774663