力扣1791. 找出星型图的中心节点
如果一个数字是中心节点的话,它的出现次数一定是等于数组的长度的。基于这个想法,可以将edges展开为一维数组,使用Counter统计每个数字出现的次数,找出出现次数等于数组的长度的,即为最终结果。
from collections import Counter
from typing import List
def findCenter(edges: List[List[int]]) -> int:
n = len(edges)
nums = [item for sublist in edges for item in sublist]
counter = Counter(nums)
for k, v in counter.items():
if v == n:
return k
edges = [[1, 2], [2, 3], [4, 2]]
print(findCenter(edges)) # 2
END