字节跳动面试题.寻找抖音红人

1.题目描述

n个人之间存在m个关系对，关系具有传递性，假如A关注B，B关注C，那么A就间接关注了C。如果一个人被除他之外的所有人都直接或间接关注，那么这个人就是抖音红人，求抖音红人的总数。

2.解题思路

将用户之间的关注关系使用一个字典表示，key为被关注的用户，value为关注此用户的用户，遍历每个key的value中的用户，补充间接关注关系，最后判断哪个用户的len(value)为用户总数，则该用户为抖音红人。

3.代码实现

# -*- 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):
        if i % 2 != 0:  # 奇数表示拿到的是关注的对象 奇数前面的偶数表示关注此对象的人
            # 把自己加入进来
            gx_dict[x].add(x)
            gx_dict[x].add(gx_list[i-1])
    # 需要多次检测，因为有间接的关注关系，一个用户最多依次被num个用户关注（包括自己），因此循环num次
    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
 
    star_num = 0
    for star, user in gx_dict.iteritems():
        if len(user) == int(num):
            star_num += 1
    return star_num
 
if __name__ == "__main__":
    N = raw_input()  
    M = raw_input()  
    Guan_xi = raw_input().split(' ')  
    print split_guanxi(Guan_xi, N)