Find the Celebrity – Python 3 (Week 10 – 17)

"""
The knows API is already defined for you.
@param a, person a
@param b, person b
@return a boolean, whether a knows b
you can call Celebrity.knows(a, b)
"""


class Solution:
    # @param {int} n a party with n people
    # @return {int} the celebrity's label or -1
    def findCelebrity(self, n):
        # Write your code here
        
        ans = 0
        
        for i in range(n):
            if Celebrity.knows(ans, i):
                ans = i
            
            
        i = 0
        for i in range(n):
            if ans != i and Celebrity.knows(ans, i):
                return -1
            if ans != i and not Celebrity.knows(i, ans):
                return -1
            
            
        return ans