For each kid, check if their current candies plus the extra candies is greater than or equal to the maximum candies any kid currently has. Return a list of booleans.
Same logic but uses an explicit loop and append() instead of a list comprehension, for clarity.
""" 1431.1 - Solution 1 - Find Max and Compare """
#####################################################################################
# Imports
#####################################################################################
from typing import List
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def kidsWithCandies(self, candies: List[int], extraCandies: int) -> List[bool]:
"""Kids With the Greatest Number of Candies Function"""
max_candies = max(candies)
return [candy + extraCandies >= max_candies for candy in candies]
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().kidsWithCandies([2, 3, 5, 1, 3], 3))
print(Solution().kidsWithCandies([4, 2, 1, 1, 2], 1))
print(Solution().kidsWithCandies([12, 1, 12], 10))
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()
""" 1431.2 - Solution 2 - Verbose Loop Approach """
#####################################################################################
# Imports
#####################################################################################
from typing import List
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def kidsWithCandies(self, candies: List[int], extraCandies: int) -> List[bool]:
"""Kids With the Greatest Number of Candies Function"""
max_candies = max(candies)
result = []
for candy in candies:
if candy + extraCandies >= max_candies:
result.append(True)
else:
result.append(False)
return result
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().kidsWithCandies([2, 3, 5, 1, 3], 3))
print(Solution().kidsWithCandies([4, 2, 1, 1, 2], 1))
print(Solution().kidsWithCandies([12, 1, 12], 10))
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()