Skip to main content

Celebrity Problem - Medium


Problem : In a party of N people, only one person is known to everyone. Such a person may be present in the party, if yes, (s)he doesn't know anyone in the party. We can only ask questions like “does A know B? Find the celebrity in minimum number of questions.

This is a very interesting problem. The answer, ie the least number of questions required, changes based on what's given. Are we guaranteed there's one celebrity in the party? Does every non-celeb know every other non-celeb? ( not very ideal, but under a best case scenario. )

Once again, the rules
  • There is at most one celebrity at the party
  • Everyone knows the celebrity.
  • The celebrity knows no one.
  • We can only ask the question "Does A know B?"

When we ask the question, we can get a "yes" or "no". Each conveying information about both the participants in the question. Let's see those outcomes...

If the Answer is "Yes", we know A is definitely not a celebrity. But B is a candidate for a celebrity.

If the Answer is "No", we know both A and B are both candidates or both are Non-Celebrities.

Turns out, we can do this in linear time. We love those kinds, don't we?

Here we simply assume the first guy is a celeb. Each time you will ask a question
Does my assumed-celeb know the current-guy ?
 There are two outcomes,

  • If the answer is "Yes", we know our assumption was wrong,we can eliminate our assumption. But the next-guy could be a celeb. So we make him our new assumption.
  • If the answer is "No", our assumption is holding strong, we can safely eliminate the current-guy.


Evidently, every time you ask the question, one guy gets eliminated from candidacy. So we'll need to ask (n - 1) questions to get one strong candidate.

At this point, if we were assured there was one celeb we can end the algorithm here. If there's a possibility, there was no celebrity, we'll have to perform one more scan to eliminate the end guy as well.

Theoretically, that's (n - 1) questions. So we need to ask a minimum of 2(n-1) questions to figure out the celebrity. I'll attempt to prove the lower bound some other time, will be handy for my Algorithm class.


import random
N = 10
#prepare testcase
R = range(N)
rand_celeb = random.randint(0,N-1)
m = {}
for i in xrange(N):
if i != rand_celeb:
m[i] = [random.randint(0,N-1) for j in xrange(10)] + [rand_celeb]
else:
m[i] = []
def knows(i,j): #i knows j?
return j in m[i]
def find_celeb(n):
celeb = 0
for i in xrange(1,n):
if not knows(i,celeb):
celeb = i
for i in xrange(n):
if i != celeb and knows(celeb,i):
return None
return celeb
print R
print "Actual Celeb : ",rand_celeb
print "Found Celeb : ",find_celeb(len(R))
Let's extend this a bit, assume there were multiple celebrities at the party (or None). (A celeb doesn't know another celeb.) Although, I don't claim this is the most efficient algorithm, it certainly works. The first part of the solution remains the same, we find a celebrity, any one will do. Notice the original algorithm will do this for you. We include one more pass at the end, asking the question "Does X know Celebrity?". If the answer is "No", we KNOW that X is another celebrity. Here's the code.

import random
N = 100
#prepare testcase
R = range(N)
rand_celeb = list(set(random.randint(0,N-1) for i in xrange(5)))
m = {}
for i in xrange(N):
if i not in rand_celeb:
m[i] = [random.randint(0,N-1) for j in xrange(10)] +rand_celeb
else:
m[i] = []
def knows(i,j): #i knows j?
return j in m[i]
def find_celebs(n):
celeb = 0
for i in xrange(1,n):
if not knows(i,celeb):
celeb = i
celebs = []
for i in xrange(n):
if i != celeb and knows(celeb,i):
return None
if not knows(i,celeb):
celebs.append(i)
return celebs
print R
print "Actual Celeb : ",rand_celeb
print "Found Celeb : ",find_celebs(len(R))
Lovely question, Thanks to CodeBunk for posting it.

Comments

Popular posts from this blog

Find Increasing Triplet Subsequence - Medium

Problem - Given an integer array A[1..n], find an instance of i,j,k where 0 < i < j < k <= n and A[i] < A[j] < A[k]. Let's start with the obvious solution, bruteforce every three element combination until we find a solution. Let's try to reduce this by searching left and right for each element, we search left for an element smaller than the current one, towards the right an element greater than the current one. When we find an element that has both such elements, we found the solution. The complexity of this solution is O(n^2). To reduce this even further, One can simply apply the longest increasing subsequence algorithm and break when we get an LIS of length 3. But the best algorithm that can find an LIS is O(nlogn) with O( n ) space . An O(nlogn) algorithm seems like an overkill! Can this be done in linear time? The Algorithm: We iterate over the array and keep track of two things. The minimum value iterated over (min) The minimum increa...

Merge k-sorted lists - Medium

Problem - Given k-sorted lists, merge them into a single sorted list. A daft way of doing this would be to copy all the list into a new array and sorting the new array. ie O(n log(n)) The naive method would be to simply perform k-way merge similar to the auxiliary method in Merge Sort. But that is reduces the problem to a minimum selection from a list of k-elements. The Complexity of this algorithm is an abysmal O(nk). Here's how it looks in Python. We maintain an additional array called Index[1..k] to maintain the head of each list. We improve upon this by optimizing the minimum selection process by using a familiar data structure, the Heap! Using a MinHeap, we extract the minimum element from a list and then push the next element from the same list into the heap, until all the list get exhausted. This reduces the Time complexity to O(nlogk) since for each element we perform O(logk) operations on the heap. An important implementation detail is we need to keep track ...

3SUM - Hard

Problem - Given an Array of integers, A. Find out if there exists a triple (i,j,k) such that A[i] + A[j] + A[k] == 0. The 3SUM  problem is very similar to the 2SUM  problem in many aspects. The solutions I'll be discussing are also very similar. I highly recommend you read the previous post first, since I'll explain only the differences in the algorithm from the previous post. Let's begin, We start with the naive algorithm. An O(n^3) solution with 3 nested loops each checking if the sum of the triple is 0. Since O(n^3) is the higher order term, we can sort the array in O(nlogn) and add a guard at the nested loops to prune of parts of the arrays. But the complexity still remains O(n^3). The code is pretty simple and similar to the naive algorithm of 2SUM. Moving on, we'll do the same thing we did in 2SUM, replace the inner-most linear search with a binary search. The Complexity now drops to O(n^2 logn) Now, the hash table method, this is strictly not ...