LeetCode 1431. Kids With the Greatest Number

Intuition

Trivial Solution

1. Track max_candies and
2. For each element in an array add the extra candy and compare with max candies
3. if greater append True to Result and if not append False

Optimal Solution

class Solution:
    def kidsWithCandies(self, candies: List[int], extraCandies: int) -> List[bool]:
        # if candies[i] + extra >= max_candies: true else: false
        max_candies = max(candies)
        res = []

        for candy in candies:
            if candy + extraCandies >= max_candies:
                res.append(True)
            else:
                res.append(False)

        return res

Time Complexity

Time Complexity O(N)

Space Complexity O(N)