题目描述
小镇里有 n 个人,按从 1 到 n 的顺序编号。传言称,这些人中有一个暗地里是小镇法官。
如果小镇法官真的存在,那么:
小镇法官不会信任任何人。
每个人(除了小镇法官)都信任这位小镇法官。
只有一个人同时满足属性 1 和属性 2 。
给你一个数组 trust ,其中 trust[i] = [ai, bi] 表示编号为 ai 的人信任编号为 bi 的人。
如果小镇法官存在并且可以确定他的身份,请返回该法官的编号;否则,返回 -1 。
题解
建图,求节点的入度和初度
1.C
int findJudge(int n, int** trust, int trustSize, int* trustColSize){
int* inDegrees = (int *)malloc(sizeof(int)*(n+1));
int* outDegrees = (int *)malloc(sizeof(int)*(n+1));
memset(inDegrees, 0, sizeof(int)*(n+1));
memset(outDegrees, 0, sizeof(int)*(n+1));
for (int i = 0; i < trustSize; ++i) {
int x = trust[i][0], y = trust[i][1];
++inDegrees[y];
++outDegrees[x];
}
for (int i = 1; i <= n; ++i) {
if (inDegrees[i] == n - 1 && outDegrees[i] == 0) {
return i;
}
}
return -1;
}
2. python
2.1计数
Counter
计数器,返回一个字典
2.2 next()
对迭代器迭代,设置default=-1
使找不到法官是返回-1
class Solution:
def findJudge(self, n: int, trust: List[List[int]]) -> int:
inDegrees = Counter(y for _, y in trust)
outDegrees = Counter(x for x, _ in trust)
return next((i for i in range(1, n+1) if inDegrees[i] == n-1 and outDegrees[i] == 0), -1)