从 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]