题目
有一个无向的 星型 图,由 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
解题
方法一:
查找出现n次的节点
class Solution { public: int findCenter(vector<vector<int>>& edges) { int n=edges.size(); unordered_map<int,int> map; for(vector<int>& edge:edges){ map[edge[0]]++; map[edge[1]]++; } for(auto it=map.begin();it!=map.end();it++){ if(it->second==n){ return it->first; } } return -1; } };
方法二:
可以任选两条边,然后寻找这两条边的公共节点,该节点即为星型图的中心节点。
class Solution { public: int findCenter(vector<vector<int>>& edges) { return edges[0][0] == edges[1][0] || edges[0][0] == edges[1][1] ? edges[0][0] : edges[0][1]; } };