题干
小镇里有 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
解法
public static int findJudge(int n, int[][] trust) {
int[] dp = new int[n + 1];
for (int[] pp : trust) {
int p1 = pp[0];
int p2 = pp[1];
if (p1 == p2) {
dp[p1] = -1;
continue;
}
dp[p1] = -1; // 失去了做法官的机会
if (dp[p2] != -1) { // 这里先假设不需要去重
dp[p2] += 1;
}
}
for (int i = 1; i < dp.length; i++) {
if (dp[i] == n - 1) { // 找到符合法官资格的
return i;
}
}
return -1;
}
想了一版本优化的,排名上没有任何提升
public static int findJudge(int n, int[][] trust) {
if (n == 1) {
return 1;
}
int[] dp = new int[n + 1];
List<Integer> anslist = new ArrayList<>();
for (int[] pp : trust) {
dp[pp[0]] = Integer.MIN_VALUE; // 失去了做法官的机会,这里直接给最小值,反正够加
dp[pp[1]] += 1;
if (dp[pp[1]] == n - 1) { // 这里只要后面不信任别人,他就是法官
anslist.add(pp[1]);
}
}
// 这里是把有法官潜力的收集起来遍历,减少遍历的数量,但是没啥效果
for (int i = 0; i < anslist.size(); i++) {
if (dp[anslist.get(i)] == n - 1) {
return anslist.get(i);
}
}
return -1;
}