题目如下,比较简单的一个题,默认用户自己关注了自己,并且间接关注具有传递性。使用Python的集合来解决会比较容易。
思路如下:将用户之间的关注关系使用一个字典表示,key为被关注的用户,value为关注此用户的用户,遍历每个key的value中的用户,补充间接关注关系,最后判断哪个用户的len(value)为用户总数,则该用户为抖音红人。
# -*- coding:utf-8 -*-
import collections
def split_guanxi(gx_list, num):
gx_dict = collections.defaultdict(set) # key表示用户,value为list,表示谁关注了他
for i, x in enumerate(gx_list):
# print i, x
if i % 2 != 0: # 奇数表示拿到的是关注的对象 奇数前面的偶数表示关注此对象的人
gx_dict[x].add(x)
gx_dict[x].add(gx_list[i-1])
#print gx_dict
for i in range(int(num)): # 需要多次检测,因为有间接的关注关系
for star, user in gx_dict.iteritems(): # star 表示被关注的人, user表示关注者
add_people = set()
for x in user:
if x in gx_dict:
add_people |= gx_dict[x]
gx_dict[star] |= add_people
#print gx_dict
star_num = 0
for star, user in gx_dict.iteritems():
#print star, len(user)
if len(user) == int(num):
star_num += 1
return star_num
if __name__ == "__main__":
#while True: # 在牛客网上提交 python 不用写循环,java要
N = raw_input() # users_num
M = raw_input() # guanxi_num
Guan_xi = raw_input().split(' ') # guanxi, 用空格分割,得到list
print split_guanxi(Guan_xi, N)