从 Python 中的关系列表中查找共同关注者的程序

2026-06-03 2 花语

假设我们有一个名为关系的列表。关系列表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]