要求
假设你是一个专业的狗仔,参加了一个 n 人派对,其中每个人被从 0 到 n - 1 标号。在这个派对人群当中可能存在一位 “名人”。所谓 “名人” 的定义是:其他所有 n - 1 个人都认识他/她,而他/她并不认识其他任何人。
现在你想要确认这个 “名人” 是谁,或者确定这里没有 “名人”。而你唯一能做的就是问诸如 “A 你好呀,请问你认不认识 B呀?” 的问题,以确定 A 是否认识 B。你需要在(渐近意义上)尽可能少的问题内来确定这位 “名人” 是谁(或者确定这里没有 “名人”)。
在本题中,你可以使用辅助函数 bool knows(a, b) 获取到 A 是否认识 B。请你来实现一个函数 int findCelebrity(n)。
派对最多只会有一个 “名人” 参加。若 “名人” 存在,请返回他/她的编号;若 “名人” 不存在,请返回 -1。
示例 1:
输入: graph = [ [1,1,0],
[0,1,0],
[1,1,1]
]
输出: 1
解释: 有编号分别为 0、1 和 2 的三个人。graph[i][j] = 1 代表编号为 i 的人认识编号为 j 的人,而 graph[i][j] = 0 则代表编号为 i 的人不认识编号为 j 的人。“名人” 是编号 1 的人,因为 0 和 2 均认识他/她,但 1 不认识任何人。
示例 2:
输入: graph = [
[1,0,1],
[1,1,0],
[0,1,1]
]
输出: -1
解释: 没有 “名人”
提示:
- n == graph.length
- n == graph[i].length
- 2 <= n <= 100
- graph[i][j] 是 0 或 1.
- graph[i][i] == 1
进阶:如果允许调用 API knows 的最大次数为 3 * n ,你可以设计一个不超过最大调用次数的解决方案吗?
核心代码
# The knows API is already defined for you.
# return a bool, whether a knows b
# def knows(a: int, b: int) -> bool:
class Solution:
def findCelebrity(self, n: int) -> int:
celebrity = 0
for i in range(1,n):
if knows(celebrity,i):
celebrity = i
for i in range(celebrity):
if knows(celebrity,i):
return -1
for i in range(n):
if not knows(i,celebrity):
return -1
return celebrity
解题思路:这个问题实际上就是在读懂问题,在第一次循环的时候,我们到celebrity(假设是k)的时候,我们可以确定的是:前k个人都不是名人,他们认识别人,第k个人和后面的所有人都不认识;第二次循环的时候我们要判断的是k和之前的k-1个人是不是认识;第三次循环,我们看看这个人认不认识其他任何一个人,如果以上的都满足,就是名人,否则不是。