题目地址:
https://www.lintcode.com/problem/find-the-celebrity/description
有
n
n
n个人参加聚会,人的编号为
0
∼
n
−
1
0\sim n-1
0∼n−1。给定一个函数boolean know(a,b)
,其返回true意味着
a
a
a认识
b
b
b。一个人是名人,意思是他不认识所有其他的人,但其他人都认识他。要求找到名人。题目保证如果存在名人,那一定唯一。如果不存在,则返回
−
1
-1
−1。
如果调用了boolean know(a,b)
并返回了true,意味着
a
a
a认识
b
b
b,所以意味着
a
a
a不是名人,而
b
b
b是名人。所以我们开一个变量,假设
0
0
0是名人,接着一遍循环,依次检验这个变量是否认识
i
i
i号人,如果认识,就将变量赋值为
i
i
i。这样一遍遍历完之后,变量里就存了潜在的名人编号,最后check一下即可。
算法正确性证明:
只需证明如果名人存在,则循环结束时变量里一定存了名人即可。在遍历名人编号之前的人的时候,无论怎样,只要走到了名人,那变量一定会存名人,接着,由于名人谁也不认识,所以变量之后不会改变。证明完毕。
代码如下:
public class Solution extends Relation {
/**
* @param n a party with n people
* @return the celebrity's label or -1
*/
public int findCelebrity(int n) {
// Write your code here
int res = 0;
for (int i = 1; i < n; i++) {
if (knows(res, i)) {
res = i;
}
}
// 最后检验一下res是否是名人
for (int i = 0; i < n; i++) {
if (res != i && knows(res, i)) {
return -1;
}
if (res != i && !knows(i, res)) {
return -1;
}
}
return res;
}
}
class Relation {
boolean knows(int a, int b) {
return false;
}
}
时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)。