Python入门习题（53）——团体程序设计天梯赛习题：帅到没朋友

题目描述

题目编号: L1-020
题目标题: 帅到没朋友
作者: 陈越
单位: 浙江大学
时间限制: 200 ms
内存限制: 64 MB
代码长度限制: 16 KB

当芸芸众生忙着在朋友圈中发照片的时候，总有一些人因为太帅而没有朋友。本题就要求你找出那些帅到没有朋友的人。

输入格式：
输入第一行给出一个正整数N（≤100），是已知朋友圈的个数；随后N行，每行首先给出一个正整数K（≤1000），为朋友圈中的人数，然后列出一个朋友圈内的所有人——为方便起见，每人对应一个ID号，为5位数字（从00000到99999），ID间以空格分隔；之后给出一个正整数M（≤10000），为待查询的人数；随后一行中列出M个待查询的ID，以空格分隔。

注意：没有朋友的人可以是根本没安装“朋友圈”，也可以是只有自己一个人在朋友圈的人。虽然有个别自恋狂会自己把自己反复加进朋友圈，但题目保证所有K超过1的朋友圈里都至少有2个不同的人。

输出格式：
按输入的顺序输出那些帅到没朋友的人。ID间用1个空格分隔，行的首尾不得有多余空格。如果没有人太帅，则输出No one is handsome。

注意：同一个人可以被查询多次，但只输出一次。

输入样例1：
3
3 11111 22222 55555
2 33333 44444
4 55555 66666 99999 77777
8
55555 44444 10000 88888 22222 11111 23333 88888

输出样例1：
10000 88888 23333

输入样例2：
3
3 11111 22222 55555
2 33333 44444
4 55555 66666 99999 77777
4
55555 44444 22222 11111

输出样例2：
No one is handsome

解题思路

1. 如何判断有朋友还是没朋友？一个人的编号出现在2个人以上的朋友圈内，则视为有朋友；否则视为没朋友。朋友圈里只有自己一个人，不算有朋友。

数据结构

1. 用friends集合存储有朋友的人。查询某个人有没有朋友，就是查询该人有没有出现在friends集合内。friends集合的类型是set，而不是list。用set类型有两点好处：（1）避免重复。往friends集合内添加一个人，该人已经在friends集合内的话，不会有实际添加效果。（2）查询效率更高。Python 中list ,set,dict的大规模查找效率一文指出，set的in操作比list的in操作快得多。
2. 题目要求：“按输入的顺序输出那些帅到没朋友的人。”，并指出：“同一个人可以被查询多次，但只输出一次。” 为此，用result_list列表按输入的顺序存储没朋友的人，用result_set集合存储已经被查询过的人。为啥要用result_set集合？这是因为判断一个人是不是被查询多次，用到result_set集合（set类型）的in操作，set的in操作比list的in操作快得多。

算法

1. 读入每一组朋友圈，把2人以上的朋友圈内人加入friends集合。
2. 对于每一个被查询的人，如果没有在friends集合内——没朋友，也没有在result_set集合内——没被查询过，则把该人加入到result_list和result_set内。
3. 如果result_list为空，那么输出“No one is handsome”；否则，输出result_list内的人的ID。

细节

1. 如果不使用result_set，而是使用result_list，那么会超时。
2. 只有一个人的朋友圈，不算有朋友，这个人不能加入到friends集合内。

参考答案

n = int(input())
#找出有朋友的人
friends = set()  #有朋友的人们
for i in range(n):
    line = input().split()
    if int(line[0]) == 1:
        continue  #朋友圈里只有自己
    for f in line[1:]:
        friends.add(f)

nq = int(input())
qs = input().split()
result_list = []  #按查询顺序从前到后存储没朋友的人
result_set = set()  #存储已被查询过的人。
for q in qs:
    if q not in friends and q not in result_set:  #result_set的in操作比result_list的in操作快多了。
        result_list.append(q)  #没朋友，且没有被查询过
        result_set.add(q)

if len(result_list) == 0:
    print("No one is handsome")
else:
    print(' '.join(result_list))

测试用例

1. 题目描述给出的第一组测试用例覆盖了找到没朋友的人的情形，也覆盖了一个人被查询多次，但只输出依次的情形。
2. 题目描述给出的第二组测试用例覆盖了输出No one is handsome的情形。
3. 朋友圈只有1个人的情形。
   样例输入
   4
   3 11111 22222 55555
   2 33333 44444
   1 88888
   4 55555 66666 99999 77777
   8
   55555 44444 10000 88888 22222 11111 23333 88888
   样例输出
   10000 88888 23333

小结

1. Python 中list ,set,dict的大规模查找效率一文指出，set的in操作比list的in操作快得多。
2. 为减少程序运行的时间开销，要仔细选择数据结构。