【图论】997. 找到小镇的法官

题目描述

(简单)小镇里有 n 个人，按从 1 到 n 的顺序编号。传言称，这些人中有一个暗地里是小镇法官。

如果小镇法官真的存在，那么：

1. 小镇法官不会信任任何人。
2. 每个人（除了小镇法官）都信任这位小镇法官。
3. 只有一个人同时满足属性 1 和属性 2 。

给你一个数组 trust ，其中 trust[i] = [ai, bi] 表示编号为 ai 的人信任编号为 bi 的人。

如果小镇法官存在并且可以确定他的身份，请返回该法官的编号；否则，返回 -1 。

示例：

输入：n = 2, trust = [[1,2]]
输出：2

解题思路

属性3指出，若存在有且只有一人满足属性1和属性2，故只需要维护每个镇民相信人的数量，以及被人信任的数量。

其实和图的联系并不强，能连系在一起的概念只有出度与入度，参考这里。至于本题，出度，即信任他人的个数；入度，即被人信任的个数。

所以遍历所有人，当发现再相同的index下，出度为 0，入度为n - 1，即为法官。

代码实现

该怎么维护出度与入度呢？考虑到本题只在意出度是否为零，非零情况下具体是多少并不在意，可以使用 unordered_set 维护出度。同时用二维数组维护入度，第一维为镇民id，第二维存信任该id镇民的其他镇民，即vector<vector<int>> judge; judge[id] = {id_1, id_2, ..., id_x};:

class Solution {
public:
    int findJudge(int n, vector<vector<int>>& trust) {
        unordered_set<int> people;
        vector<vector<int>> judge(n);
        for(auto vec : trust){
            people.insert(vec[0]);
            judge[vec[1] - 1].push_back(vec[0]);
        }

        for(int i = 0; i < n; i++){
            if(!people.count(i + 1) && judge[i].size() == n - 1) return i + 1;
        }
        return -1;
    }
};

运行结果：

啊这，可以考虑优化优化。这个二维数组添加元素的操作背后涉及数组扩容、拷贝，复杂度为O(n)。考虑将其优化掉，我们其实并不关心谁信任谁，或者谁被谁信任。在意的是其具体数值，故可以使用一维数组维护个数：

class Solution {
public:
    int findJudge(int n, vector<vector<int>>& trust) {
        //入度
        vector<int> in(n);
        //出度
        vector<int> out(n);

        for(auto vec : trust){
            in[vec[1] - 1]++;
            out[vec[0] - 1]++;
        }

        for(int i = 0; i < n; i++){
            if(in[i] == n - 1 && out[i] == 0) return i + 1;
        }

        return -1;
    }
};

运行结果：

啊这。。。再试试！换引用！

class Solution {
public:
    int findJudge(int n, vector<vector<int>>& trust) {
        //入度
        vector<int> in(n);
        //出度
        vector<int> out(n);
		//注意，循环时原本的局部变量使用引用替换
        for(auto& vec : trust){
            in[vec[1] - 1]++;
            out[vec[0] - 1]++;
        }

        for(int i = 0; i < n; i++){
            if(in[i] == n - 1 && out[i] == 0) return i + 1;
        }

        return -1;
    }
};

性能提升：

哇偶！