leetcode 997. 找到小镇的法官(Find the Town Judge)

题目描述:

在一个小镇里,按从 1N 标记了 N 个人。传言称,这些人中有一个是小镇上的秘密法官。

如果小镇的法官真的存在,那么:

  1. 小镇的法官不相信任何人。
  2. 每个人(除了小镇法官外)都信任小镇的法官。
  3. 只有一个人同时满足属性 1 和属性 2 。

给定数组 trust,该数组由信任对 trust[i] = [a, b] 组成,表示标记为 a 的人信任标记为 b 的人。

如果小镇存在秘密法官并且可以确定他的身份,请返回该法官的标记。否则,返回 -1

示例 1:

输入:N = 2, trust = [[1,2]]
输出:2

示例 2:

输入:N = 3, trust = [[1,3],[2,3]]
输出:3

示例 3:

输入:N = 3, trust = [[1,3],[2,3],[3,1]]
输出:-1

示例 4:

输入:N = 3, trust = [[1,2],[2,3]]
输出:-1

示例 5:

输入:N = 4, trust = [[1,3],[1,4],[2,3],[2,4],[4,3]]
输出:3

提示:

  1. 1 <= N <= 1000
  2. trust.length <= 10000
  3. trust[i] 是完全不同的
  4. trust[i][0] != trust[i][1]
  5. 1 <= trust[i][0], trust[i][1] <= N

解法:

class Solution {
public:
    int findJudge(int N, vector<vector<int>>& trust) {
        vector<bool> flag(N+1, true);
        vector<int> vals(N+1, 0);
        for(vector<int> tst : trust){
            flag[tst[0]] = false;
            vals[tst[1]]++;
        }
        for(int i = 1; i <= N; i++){
            if(flag[i] && vals[i] == N-1){
                // check i trust others
                return i;
            }
        }
        return -1;
    }
};

猜你喜欢

转载自www.cnblogs.com/zhanzq/p/10674289.html