Leetcode 997:找到小镇的法官(超详细的解法!!!)

在一个小镇里,按从 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

解题思路

这个问题非常简单,我们可以建立一个defaultdict(set),然后遍历trust中的每个元素it(将it[0]作为key,将it[1]作为value)添加到dict中,这样我们就建立了一个信任关系的映射。我们知道每个人(除了小镇法官外)都信任小镇的法官,那么我们只需要查找dict中是不是有一个value的长度等于N-1,如果存在的话,那么对应的key就是法官。此时还有一个问题,就是小镇的法官不相信任何人,我们只需要新建一个set,将itkey添加到set中,如果我们返回的数在set中存在的话,那么说明存在法官相信他人的情况

这里还有一个边界条件没有考虑,就是小镇只有一个人的时候,那么这个人一定是法官。

from collections import defaultdict
class Solution:
    def findJudge(self, N: int, trust: List[List[int]]) -> int:
        if not trust:
            return 1
        trust_map = defaultdict(set)
        trust_set = set()
        for it in trust:
            trust_map[it[1]].add(it[0])
            trust_set.add(it[0])
        
        for key, val in trust_map.items():
            if len(val) == N-1 and key not in trust_set:
                return key
        return -1

我们还有一个更加出色的想法,就是建立有向图。将小镇里的每个人当成一个节点,那么对于法官这个节点来说,指向它的边的数量减去它指向的边的数量一定是N-1,可以理解为七桥问题。(边表示为it[0]指向it[1]

class Solution:
    def findJudge(self, N: int, trust: List[List[int]]) -> int:
        count = [0] * (N + 1)
        for i, j in trust:
            count[i] -= 1
            count[j] += 1
        for i in range(1, N + 1):
            if count[i] == N - 1:
                return i
        return -1

reference:

https://leetcode.com/problems/find-the-town-judge/discuss/242938/JavaC%2B%2BPython-Directed-Graph

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题,希望大家指出!!!

你可能感兴趣的:(Problems,leetcode解题指南,leetcode)