LeetCode 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 是中心节点。

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

由于星型图的两条边必定有相同节点,且相同节点就是中心点,因此找两条边的相同节点即可:

class Solution {
public:
    int findCenter(vector<vector<int>>& edges) {
        int nodeNum = edges.size() + 1;
        if (nodeNum < 3) {
            return -1;
        }

        return edges[0][0] == edges[1][0] || edges[0][0] == edges[1][1] ? edges[0][0] : edges[0][1];
    }
};

如果节点数为n,此解法时间复杂度O(1),空间复杂度O(1)。

posted @ 2023-01-26 23:09  epiphanyy  阅读(20)  评论(0)    收藏  举报  来源