python小课——零基础入门——学习笔记Leetcode模拟面试LeetCode刷题

LeetCode 1791. 找出星型图的中心节点

2021-09-27  本文已影响0人  freesan44

题目

有一个无向的 星型 图,由 n 个编号从 1 到 n 的节点组成。星型图有一个 中心 节点,并且恰有 n - 1 条边将中心节点与其他每个节点连接起来。

给你一个二维整数数组 edges ,其中 edges[i] = [ui, vi] 表示在节点 ui 和 vi 之间存在一条边。请你找出并返回 edges 所表示星型图的中心节点。

示例 1:


输入:edges = [[1,2],[2,3],[4,2]]
输出:2
解释:如上图所示,节点 2 与其他每个节点都相连,所以节点 2 是中心节点。
示例 2:

输入:edges = [[1,2],[5,1],[1,3],[1,4]]
输出:1

提示:

3 <= n <= 105
edges.length == n - 1
edges[i].length == 2
1 <= ui, vi <= n
ui != vi
题目数据给出的 edges 表示一个有效的星型图

解题思路

class Solution:
    def findCenter(self, edges: List[List[int]]) -> int:
        resList = []
        #把全部数组拆解出来形成一个List,然后用Counter归纳,找出数量为n的
        for i in edges:
            resList += i
        # print(resList)
        from collections import Counter
        count = Counter(resList)
        for (key, val) in count.items():
            # print(key,val)
            # print(len(edges)-1)
            if val == (len(edges)):
                return key
        return 0

if __name__ == '__main__':
    edges = [[1,2],[2,3],[4,2]]
    ret = Solution().findCenter(edges)
    print(ret)
上一篇 下一篇

猜你喜欢

热点阅读