The Celebrity Problem - GeeksforGeeks

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 stranger (celebrity) in minimum number of questions.

We can describe the problem input as an array of numbers/characters representing persons in the party. We also have a hypothetical function HaveAcquaintance(A, B) which returns true if A knows B, false otherwise. How can we solve the problem.

Recommended: Please solve it on “PRACTICE” first, before moving on to the solution.

We measure the complexity in terms of calls made to HaveAcquaintance().

Method 1 (Graph)

We can model the solution using graphs. Initialize indegree and outdegree of every vertex as 0. If A knows B, draw a directed edge from A to B, increase indegree of B and outdegree of A by 1. Construct all possible edges of the graph for every possible pair [i, j]. We have N_C₂ pairs. If celebrity is present in the party, we will have one sink node in the graph with outdegree of zero, and indegree of N-1. We can find the sink node in (N) time, but the overall complexity is O(N²) as we need to construct the graph first.

Method 2 (Recursion)

We can decompose the problem into combination of smaller instances. Say, if we know celebrity of N-1 persons, can we extend the solution to N? We have two possibilities, Celebrity(N-1) may know N, or N already knew Celebrity(N-1). In the former case, N will be celebrity if N doesn’t know anyone else. In the later case we need to check that Celebrity(N-1) doesn’t know N.

Solve the problem of smaller instance during divide step. On the way back, we find the celebrity (if present) from the smaller instance. During combine stage, check whether the returned celebrity is known to everyone and he doesn’t know anyone. The recurrence of the recursive decomposition is,

T(N) = T(N-1) + O(N)

T(N) = O(N²). You may try writing pseudo code to check your recursion skills.

Method 3 (Using Stack)
The graph construction takes O(N²) time, it is similar to brute force search. In case of recursion, we reduce the problem instance by not more than one, and also combine step may examine M-1 persons (M – instance size).

We have following observation based on elimination technique (Refer Polya’s How to Solve It book).

If A knows B, then A can’t be celebrity. Discard A, and B may be celebrity.
If A doesn’t know B, then B can’t be celebrity. Discard B, and A may be celebrity.
Repeat above two steps till we left with only one person.
Ensure the remained person is celebrity. (Why do we need this step?)

We can use stack to verity celebrity.

Push all the celebrities into a stack.
Pop off top two persons from the stack, discard one person based on return status of HaveAcquaintance(A, B).
Push the remained person onto stack.
Repeat step 2 and 3 until only one person remains in the stack.
Check the remained person in stack doesn’t have acquaintance with anyone else.

We will discard N elements utmost (Why?). If the celebrity is present in the party, we will call HaveAcquaintance() 3(N-1) times. Here is code using stack.

C++

// C++ program to find celebrity 
#include <bits/stdc++.h> 
#include <list> 
using namespace std; 
  
// Max # of persons in the party 
#define N 8 
  
// Person with 2 is celebrity 
bool MATRIX[N][N] = {{0, 0, 1, 0}, 
                    {0, 0, 1, 0}, 
                    {0, 0, 0, 0}, 
                    {0, 0, 1, 0}}; 
  
bool knows(int a, int b) 
{ 
    return MATRIX[a][b]; 
} 
  
// Returns -1 if celebrity 
// is not present. If present, 
// returns id (value from 0 to n-1). 
int findCelebrity(int n) 
{ 
    // Handle trivial  
    // case of size = 2 
  
    stack<int> s; 
  
    int C; // Celebrity 
  
    // Push everybody to stack 
    for (int i = 0; i < n; i++) 
        s.push(i); 
  
    // Extract top 2 
    int A = s.top(); 
    s.pop(); 
    int B = s.top(); 
    s.pop(); 
  
    // Find a potential celevrity 
    while (s.size() > 1) 
    { 
        if (knows(A, B)) 
        { 
            A = s.top(); 
            s.pop(); 
        } 
        else
        { 
            B = s.top(); 
            s.pop(); 
        } 
    } 
  
    // Potential candidate? 
    C = s.top(); 
    s.pop(); 
  
    // Last candidate was not  
    // examined, it leads one  
    // excess comparison (optimize) 
    if (knows(C, B)) 
        C = B; 
  
    if (knows(C, A)) 
        C = A; 
  
    // Check if C is actually 
    // a celebrity or not 
    for (int i = 0; i < n; i++) 
    { 
        // If any person doesn't  
        // know 'a' or 'a' doesn't  
        // know any person, return -1 
        if ( (i != C) && 
                (knows(C, i) ||  
                 !knows(i, C)) ) 
            return -1; 
    } 
  
    return C; 
} 
  
// Driver code 
int main() 
{ 
    int n = 4; 
    int id = findCelebrity(n); 
    id == -1 ? cout << "No celebrity" : 
               cout << "Celebrity ID " << id; 
    return 0; 
} 

Java

// Java program to find celebrity using 
// stack data structure 
  
import java.util.Stack; 
  
class GFG  
{ 
    // Person with 2 is celebrity 
    static int MATRIX[][] = { { 0, 0, 1, 0 }, 
                            { 0, 0, 1, 0 }, 
                            { 0, 0, 0, 0 },  
                            { 0, 0, 1, 0 } }; 
  
    // Returns true if a knows  
    // b, false otherwise 
    static boolean knows(int a, int b)  
    { 
        boolean res = (MATRIX[a][b] == 1) ?  
                                     true :  
                                     false; 
        return res; 
    } 
  
    // Returns -1 if celebrity  
    // is not present. If present, 
    // returns id (value from 0 to n-1). 
    static int findCelebrity(int n)  
    { 
        Stack<Integer> st = new Stack<>(); 
        int c; 
  
        // Step 1 :Push everybody 
        // onto stack 
        for (int i = 0; i < n; i++)  
        { 
            st.push(i); 
        } 
  
        while (st.size() > 1)  
        { 
            // Step 2 :Pop off top 
            // two persons from the  
            // stack, discard one  
            // person based on return 
            // status of knows(A, B). 
            int a = st.pop(); 
            int b = st.pop(); 
  
            // Step 3 : Push the  
            // remained person onto stack. 
            if (knows(a, b))  
            { 
                st.push(b); 
            } 
  
            else
                st.push(a); 
        } 
  
        c = st.pop(); 
  
        // Step 5 : Check if the last  
        // person is celebrity or not 
        for (int i = 0; i < n; i++)  
        { 
            // If any person doesn't 
            //  know 'c' or 'a' doesn't  
            // know any person, return -1 
            if (i != c && (knows(c, i) ||  
                          !knows(i, c))) 
                return -1; 
        } 
        return c; 
    } 
  
    // Driver Code 
    public static void main(String[] args)  
    { 
        int n = 4; 
        int result = findCelebrity(n); 
        if (result == -1)  
        { 
            System.out.println("No Celebrity"); 
        }  
        else
            System.out.println("Celebrity ID " +  
                                        result); 
    } 
} 
  
// This code is contributed  
// by Rishabh Mahrsee 

Output :

Celebrity ID 2

Complexity O(N). Total comparisons 3(N-1). Try the above code for successful MATRIX {{0, 0, 0, 1}, {0, 0, 0, 1}, {0, 0, 0, 1}, {0, 0, 0, 1}}.

Note: You may think that why do we need a new graph as we already have access to input matrix. Note that the matrix MATRIX used to help the hypothetical function HaveAcquaintance(A, B), but never accessed via usual notation MATRIX[i, j]. We have access to the input only through the function HaveAcquiantance(A, B). Matrix is just a way to code the solution. We can assume the cost of hypothetical function as O(1).

If still not clear, assume that the function HaveAcquiantance accessing information stored in a set of linked lists arranged in levels. List node will have next and nextLevel pointers. Every level will have N nodes i.e. an N element list, next points to next node in the current level list and the nextLevel pointer in last node of every list will point to head of next level list. For example the linked list representation of above matrix looks like,

L0 0->0->1->0
             |
L1           0->0->1->0
                       |
L2                     0->0->1->0
                                 |
L3                               0->0->1->0

The function HaveAcquanintance(i, j) will search in the list for j-th node in the i-th level. Out goal is to minimize calls to HaveAcquanintance function.

Method 4 (Using two Pointers)

The idea is to use two pointers, one from start and one from the end. Assume the start person is A, and the end person is B. If A knows B, then A must not be the celebrity. Else, B must not be the celebrity. We will find a celebrity candidate at the end of the loop. Go through each person again and check whether this is the celebrity. Below is C++ implementation.

C++

// C++ program to find  
// celebrity in O(n) time 
// and O(1) extra space 
#include <bits/stdc++.h> 
using namespace std; 
  
// Max # of persons in the party 
#define N 8 
  
// Person with 2 is celebrity 
bool MATRIX[N][N] = {{0, 0, 1, 0}, 
                     {0, 0, 1, 0}, 
                     {0, 0, 0, 0}, 
                     {0, 0, 1, 0} 
}; 
  
bool knows(int a, int b) 
{ 
    return MATRIX[a][b]; 
} 
  
// Returns id of celebrity 
int findCelebrity(int n) 
{ 
    // Initialize two pointers  
    // as two corners 
    int a = 0; 
    int b = n - 1; 
  
    // Keep moving while  
    // the two pointers 
    // don't become same.  
    while (a < b) 
    { 
        if (knows(a, b)) 
            a++; 
        else
            b--; 
    } 
  
    // Check if a is actually 
    // a celebrity or not 
    for (int i = 0; i < n; i++) 
    { 
        // If any person doesn't  
        // know 'a' or 'a' doesn't 
        // know any person, return -1 
        if ( (i != a) && 
                (knows(a, i) ||  
                !knows(i, a)) ) 
            return -1; 
    } 
  
    return a; 
} 
  
// Driver code 
int main() 
{ 
    int n = 4; 
    int id = findCelebrity(n); 
    id == -1 ? cout << "No celebrity" : 
               cout << "Celebrity ID " 
                    << id; 
    return 0; 
} 

Java

// Java program to find  
// celebrity using two  
// pointers 
  
class GFG  
{ 
    // Person with 2 is celebrity 
    static int MATRIX[][] = { { 0, 0, 1, 0 }, 
                               { 0, 0, 1, 0 },  
                              { 0, 0, 0, 0 }, 
                              { 0, 0, 1, 0 } }; 
  
    // Returns true if a knows 
    // b, false otherwise 
    static boolean knows(int a, int b)  
    { 
        boolean res = (MATRIX[a][b] == 1) ?  
                                     true :  
                                     false; 
        return res; 
    } 
  
    // Returns -1 if celebrity  
    // is not present. If present,  
    // returns id (value from 0 to n-1). 
    static int findCelebrity(int n)  
    { 
        // Initialize two pointers  
        // as two corners 
        int a = 0; 
        int b = n - 1; 
          
        // Keep moving while  
        // the two pointers 
        // don't become same. 
        while (a < b)  
        { 
            if (knows(a, b)) 
                a++; 
            else
                b--; 
        } 
  
        // Check if a is actually  
        // a celebrity or not 
        for (int i = 0; i < n; i++)  
        { 
            // If any person doesn't  
            // know 'a' or 'a' doesn't 
            // know any person, return -1 
            if (i != a && (knows(a, i) ||  
                           !knows(i, a))) 
                return -1; 
        } 
        return a; 
    } 
  
    // Driver Code 
    public static void main(String[] args)  
    { 
        int n = 4; 
        int result = findCelebrity(n); 
        if (result == -1)  
        { 
            System.out.println("No Celebrity"); 
        }  
        else
            System.out.println("Celebrity ID " +  
                                        result); 
    } 
} 
  
// This code is contributed by Rishabh Mahrsee 

C#

// C# program to find  
// celebrity using two  
// pointers 
using System; 
  
class GFG  
{ 
    // Person with 2 is celebrity 
    static int [,]MATRIX = {{ 0, 0, 1, 0 }, 
                            { 0, 0, 1, 0 },  
                            { 0, 0, 0, 0 }, 
                            { 0, 0, 1, 0 }}; 
  
    // Returns true if a knows 
    // b, false otherwise 
    static bool knows(int a, int b)  
    { 
        bool res = (MATRIX[a, b] == 1) ?  
                                  true :  
                                  false; 
        return res; 
    } 
  
    // Returns -1 if celebrity  
    // is not present. If present,  
    // returns id (value from 0 to n-1). 
    static int findCelebrity(int n)  
    { 
        // Initialize two pointers  
        // as two corners 
        int a = 0; 
        int b = n - 1; 
          
        // Keep moving while  
        // the two pointers 
        // don't become same. 
        while (a < b)  
        { 
            if (knows(a, b)) 
                a++; 
            else
                b--; 
        } 
  
        // Check if a is actually  
        // a celebrity or not 
        for (int i = 0; i < n; i++)  
        { 
            // If any person doesn't  
            // know 'a' or 'a' doesn't 
            // know any person, return -1 
            if (i != a && (knows(a, i) ||  
                          !knows(i, a))) 
                return -1; 
        } 
        return a; 
    } 
  
    // Driver Code 
    public static void Main()  
    { 
        int n = 4; 
        int result = findCelebrity(n); 
        if (result == -1)  
        { 
            Console.WriteLine("No Celebrity"); 
        }  
        else
            Console.WriteLine("Celebrity ID " +  
                                       result); 
    } 
} 
  
// This code is contributed by anuj_67. 

PHP

<?php 
// PHP program to find  
// celebrity in O(n) time 
// and O(1) extra space 
  
  
// Max # of persons  
// in the party $N = 8; 
  
// Person with 2 is celebrity 
$MATRIX = array(array(0, 0, 1, 0), 
                array(0, 0, 1, 0), 
                array(0, 0, 0, 0), 
                array(0, 0, 1, 0)); 
  
function knows( $a, $b) 
{ 
    global $MATRIX; 
    return $MATRIX[$a][$b]; 
} 
  
// Returns id of celebrity 
function findCelebrity( $n) 
{ 
    // Initialize two  
    // pointers as two corners 
    $a = 0; 
    $b = $n - 1; 
  
    // Keep moving while  
    // the two pointers 
    // don't become same.  
    while ($a < $b) 
    { 
        if (knows($a, $b)) 
            $a++; 
        else
            $b--; 
    } 
  
    // Check if a is actually 
    // a celebrity or not 
    for ( $i = 0; $i < $n; $i++) 
    { 
        // If any person doesn't  
        // know 'a' or 'a' doesn't  
        // know any person, return -1 
        if ( ($i != $a) and
                (knows($a, $i) ||  
                !knows($i, $a)) ) 
            return -1; 
    } 
  
    return $a; 
} 
  
// Driver code 
$n = 4; 
$id = findCelebrity($n); 
if($id == -1) 
echo "No celebrity" ; 
else
echo "Celebrity ID " , $id; 
  
// This code is contributed by anuj_67. 
?> 

Output :

Celebrity ID 2

Thanks to Sissi Peng for suggesting this method.

Related Article:
Number of sink nodes in a graph

Exercises:
1. Write code to find celebrity. Don’t use any data structures like graphs, stack, etc… you have access to N and HaveAcquaintance(int, int) only.

2. Implement the algorithm using Queues. What is your observation? Compare your solution with Finding Maximum and Minimum in an array and Tournament Tree. What are minimum number of comparisons do we need (optimal number of calls to HaveAcquaintance())?

— Venki. Please write comments if you find anything incorrect, or you want to share more information about the topic discussed above.

My Personal Notes

Recommended: Please solve it on “PRACTICE” first, before moving on to the solution.

C++

Java

C++

Java

C#

PHP

Recommended Posts: