PAT乙级 1065. 单身狗(25) by Python

1065. 单身狗(25)

时间限制

300 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CHEN, Yue

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

输入格式：

输入第一行给出一个正整数N（<=50000），是已知夫妻/伴侣的对数；随后N行，每行给出一对夫妻/伴侣——为方便起见，每人对应一个ID号，为5位数字（从00000到99999），ID间以空格分隔；之后给出一个正整数M（<=10000），为参加派对的总人数；随后一行给出这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


Python源码：

sum_couple = int(input())
couples = []
for i in range(sum_couple):
  name = input()
  a = name.split()[0]
  b = name.split()[1]
  c = ((a,b),(b,a))
  couples.extend(c)
dcouples=dict(couples)

sum_guest = int(input())
guest = []
str = input()
for i in range(sum_guest):
  guest.append(str.split()[i])

sum_single = 0
single = []
for i in range(sum_guest):
  if guest[i] in dcouples:
    x=dcouples.get(guest[i])
    if x in guest:
      sum_single-=1
    else:
      single.append(guest[i])
  else:
    single.append(guest[i])

print(len(single))
if len(single)>0:
  single.sort()
  print(single[0],end='')
  if len(single)>1:
    for i in range(1,len(single)):
      print(" ",end='')
      print(single[i],end='')

结果：

测试点3 4 运行超时，未知错误