从 Python 中的关系列表中查找共同关注者的程序
假设我们有一个名为关系的列表。关系列表relations[i]中的每个元素都包含两个数字[ai,bi],它表示人ai在社交媒体平台上关注bi。我们必须找到关注某人的人的列表,然后他们又关注了他们,我们必须按排序顺序返回它。
因此,如果输入类似于关系=[[0,2],[2,3],[2,0],[1,0]],那么输出将是[0,2]。
示例
让我们看下面的实现来更好地理解
def solve(relations): ans = set() seen = set() for a, b in relations: seen.add((a, b)) if (b, a) in seen: ans.add(b) ans.add(a) k = list(ans) rtr = sorted(k) return rtr relations = [ [0, 2], [2, 3], [2, 0], [1, 0] ] print(solve(relations))
输入
[[0, 2],[2, 3],[2, 0],[1, 0]]输出结果
[0, 2]