leetcode 997. 找到小镇的法官

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

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

小镇的法官不相信任何人。
每个人(除了小镇法官外)都信任小镇的法官。
只有一个人同时满足属性 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

法官是这样一个点:出度为0,并且入度为N-1

class Solution {
public:
    int findJudge(int N, vector<vector<int>>& trust) {
        vector<int> nums1(N+1,0);
        vector<int> nums2(N+1,0);
        for(int i=0;i<trust.size();i++){
            nums1[trust[i][1]]++;
            nums2[trust[i][0]]++;
        }
        int res=-1;
        for(int i=1;i<=N;i++)
            if(nums1[i]==N-1&&nums2[i]==0)  
                return i;
        return -1;
    }
};

反思代码

int res=-1;
        for(int i=1;i<=N;i++)
            if(nums1[i]==N-1)
                res = i;
        if(nums2[res]==0)  这个res可能为 -1 就越界。。写代码 要细心
            return res;
发布了124 篇原创文章 · 获赞 47 · 访问量 7万+

猜你喜欢

转载自blog.csdn.net/ludan_xia/article/details/104877145