Leecode 1791. 找出星型图的中心节点
题目描述
有一个无向的 星型 图,由 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,则中心点的度为N - 1。
所以只需要求出每个顶点的度,然后找到度为N-1的顶点,该顶点即为中心顶点。
习题代码
public int findCenter(int[][] edges) {
//用于存储图的顶点和对应顶点的度
HashMap<Integer, Integer> map = new HashMap<>();
for (int[] edge : edges) {
//将每个结点以及其出现的次数,即该节点的值和该节点的度存入哈希表
map.put(edge[0], map.getOrDefault(edge[0], 0) + 1);
map.put(edge[1], map.getOrDefault(edge[1], 0) + 1);
}
int n = map.keySet().size();
for (Integer key : map.keySet()) {
//找到度为n-1的节点
if (map.get(key) == n - 1) {
return key;
}
}
return -1;
}
- 时间复杂度:O(N + E),遍历了边的集合和顶点集合
- 空间复杂度:O(N),用到了顶点长度的哈希表