-
Notifications
You must be signed in to change notification settings - Fork 0
/
kids-with-the-greatest-number-of-candies.py
42 lines (28 loc) · 1.54 KB
/
kids-with-the-greatest-number-of-candies.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# URL : https://leetcode.com/problems/kids-with-the-greatest-number-of-candies/
# Example 1:
# Input: candies = [2,3,5,1,3], extraCandies = 3
# Output: [true,true,true,false,true]
# Explanation:
# Kid 1 has 2 candies and if he or she receives all extra candies (3) will have 5 candies --- the greatest number of candies among the kids.
# Kid 2 has 3 candies and if he or she receives at least 2 extra candies will have the greatest number of candies among the kids.
# Kid 3 has 5 candies and this is already the greatest number of candies among the kids.
# Kid 4 has 1 candy and even if he or she receives all extra candies will only have 4 candies.
# Kid 5 has 3 candies and if he or she receives at least 2 extra candies will have the greatest number of candies among the kids.
# Example 2:
# Input: candies = [4,2,1,1,2], extraCandies = 1
# Output: [true,false,false,false,false]
# Explanation: There is only 1 extra candy, therefore only kid 1 will have the greatest number of candies among the kids regardless of who takes the extra candy.
# Example 3:
# Input: candies = [12,1,12], extraCandies = 10
# Output: [true,false,true]
# Constraints:
# 2 <= candies.length <= 100
# 1 <= candies[i] <= 100
# 1 <= extraCandies <= 50
class Solution:
def kidsWithCandies(self, candies: list, extraCandies) -> list:
return list(map(lambda x : (x+extraCandies) >= max(candies), candies))
sol = Solution()
print(sol.kidsWithCandies([2,3,5,1,3], 3))
print(sol.kidsWithCandies([4,2,1,1,2], 1))
print(sol.kidsWithCandies([12,1,12], 10))