字节跳动——寻找抖音红人的笔试题

题目如下，比较简单的一个题，默认用户自己关注了自己，并且间接关注具有传递性。使用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)