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

题目来源

• LEETCODE

题目解析

题目描述

第一个版本

思路：中心节点必然连接所有的节点

class Solution {
public:
    int findCenter(vector<vector<int>>& edges) {
        std::vector<int> vec(2);
        vec[0] = edges[0][0]; vec[1] = edges[0][1];
        int size = edges.size(), ans;
        for (int i = 1; i < size; ++i) {
            int a = edges[i][0], b = edges[i][1];
            if(a == vec[0]){
                vec[1] = b;
                ans = a;
            }else if(a == vec[1]){
                vec[0] = b;
                ans = a;
            }else if(b == vec[0]){
                vec[1] = a;
                ans = b;
            }else if(b == vec[1]){
                vec[0] = a;
                ans = b;
            }
        }
        
        return ans;
    }
};

第二个版本

根据题意，中心节点必然出现在所有的 edges[i] 中，因此使用前两条边即可确定答案。

起始让 edges[0][0]和 edges[0][1]作为答案候选，然后在 edges[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];
    }
};