LeetCode 0997.找到小镇的法官：计数

【LetMeFly】997.找到小镇的法官：计数

力扣题目链接：https://leetcode.cn/problems/find-the-town-judge/

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

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

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

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

如果小镇法官存在并且可以确定他的身份，请返回该法官的编号；否则，返回 -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

 

提示：

• 1 <= n <= 1000
• 0 <= trust.length <= 104
• trust[i].length == 2
• trust 中的所有trust[i] = [ai, bi] 互不相同
• ai != bi
• 1 <= ai, bi <= n

解题方法：计数

使用inDegree数组记录每个元素分别被信任几次，使用outDegree数组记录每个元素分别信任他人几次。

遍历一遍trust数组即可得到inDegree数组和outDegree数组。

遍历每个元素，若inDegree值等于$n-1$且outDegree数组为空，则返回该元素。

否则返回-1。

• 时间复杂度$O(n+len(trust))$
• 空间复杂度$O(n)$

AC代码

C++

class Solution {
public:
    int findJudge(int n, vector<vector<int>>& trust) {
        vector<int> inDegree(n + 1), outDegree(n + 1);
        for (vector<int>& v : trust) {
            inDegree[v[1]]++, outDegree[v[0]]++;
        }
        for (int i = 1; i <= n; i++) {
            if (inDegree[i] == n - 1 && outDegree[i] == 0) {
                return i;
            }
        }
        return -1;
    }
};

class Solution {
public:
    int findJudge(int n, vector<vector<int>>& trust) {
        vector<bool> cannotHim(n);
        vector<int> cntHim(n);
        for (vector<int>& v : trust) {
            cannotHim[v[0] - 1] = true;
            cntHim[v[1] - 1]++;
        }
        for (int i = 0; i < n; i++) {
            if (!cannotHim[i] && cntHim[i] == n - 1) {
                return i + 1;
            }
        }
        return -1;
    }
};

Python

from typing import List
from collections import Counter

class Solution:
    def findJudge(self, n: int, trust: List[List[int]]) -> int:
        inDegree = Counter(y for x, y in trust)
        outDegree = Counter(x for x, y in trust)
        for th in range(n):
            if inDegree[th + 1] == n - 1 and not outDegree[th + 1]:
                return th + 1
        return -1

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

Tisfy：https://letmefly.blog.csdn.net/article/details/142446520