LeetCode 997. 找到小镇的法官（C++、python）

在一个小镇里，按从 1 到 N 标记了 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 <= N <= 1000

trust.length <= 10000

trust[i] 是完全不同的

trust[i][0] != trust[i][1]

1 <= trust[i][0], trust[i][1] <= N

C++

class Solution {
public:
    int findJudge(int N, vector<vector<int>>& trust) 
    {
        int m=trust.size();
        if(0==m)
        {
            if(1==N)
            {
                return 1;
            }
        }
        map<int,int> tmp;
        map<int,int> count;
        for(int i=0;i<m;i++)
        {
            tmp[trust[i][0]]=1;
        }
        for(int i=0;i<m;i++)
        {
            if(tmp[trust[i][1]]!=1)
            {
                count[trust[i][1]]++;
            }
        }
        for(auto it:count)
        {
            if(it.second==N-1)
            {
                return it.first;
            }
        }
        return -1;
    }
};

python

class Solution:
    def findJudge(self, N: int, trust: List[List[int]]) -> int:
        m=len(trust)
        if 1==N:
            if 0==m:
                return 1
        tmp={}
        dic={}
        for i in range(m):
            tmp[trust[i][0]]=1
        for i in range(m):
            if trust[i][1] not in tmp:
                if trust[i][1] not in dic:
                    dic[trust[i][1]]=1
                else:
                    dic[trust[i][1]]+=1
        for key in dic:
            if N-1==dic[key]:
                return key
        return -1