携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第29天,点击查看活动详情
题目描述
小镇里有 n
个人,按从 1
到 n
的顺序编号。传言称,这些人中有一个暗地里是小镇法官。
如果小镇法官真的存在,那么:
- 小镇法官不会信任任何人。
- 每个人(除了小镇法官)都信任这位小镇法官。
- 只有一个人同时满足属性 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 <= 10^4
- trust[i].length == 2
- trust 中的所有trust[i] = [ai, bi] 互不相同
- ai != bi
- 1 <= ai, bi <= n
思路分析
- 设定数组记录每个村民的状态
- 被信任的暂时先认为是法官,并用负数记录每个潜在法官出现的次数
- 状态是负数或者未投票的如果出现在投票位置则认为其身份是贫民,并对贫民进行负计数
- 对n个人的状态遍历,只有状态是负数 && 2中的出现次数和贫民的个数相等时才认为可能是法官,如果出现多个法官返回-1
AC 代码
class Solution {
public:
int findJudge(int n, vector<vector<int>>& trust) {
vector<int> person(n+1, 0);
int _size = trust.size(), notjudge = 0;
if(_size == 0 && n==1) return 1;
for(int i=0; i<_size; i++){
if(person[trust[i][1]] <= 0) person[trust[i][1]] -= 1;
if(person[trust[i][0]] <= 0){
notjudge--;
person[trust[i][0]] = 1;
}
}
int ans = -1;
for(int i= 0; i <= n; i++){
if(person[i] >= 0) continue;
if(person[i] == notjudge && ans == -1) ans = i;
else return -1;
}
return ans;
}
};
参考
LeetCode997-找到小镇的法官 - 找到小镇的法官 - 力扣(LeetCode)
python字典+双循环(python双循环过了,那这题意义在哪。。) - 找到小镇的法官 - 力扣(LeetCode)