[leetCode] 1431. Kids With the Greatest Number of Candies (Python)

▼ 문제 바로가기 (링크) ▼

https://leetcode.com/problems/kids-with-the-greatest-number-of-candies/

Kids With the Greatest Number of Candies - LeetCode

---

정수 배열이 주어진다. 각 원소에 변수 extraCandies를 더했을 때,

전체 원소 중 최대값인지 여부를 판단하고 그 불리언 값을 리스트로 반환하는 문제.

 

class Solution:
    def kidsWithCandies(self, candies: List[int], extraCandies: int) -> List[bool]:

        answer = []
        
        for i in candies:            
            i += extraCandies
            
            if i >= max(candies) :
                answer.append(True)
                i -= extraCandies    # 추가했던 값 초기화
            else:
                answer.append(False)
                i -= extraCandies     # 추가했던 값 초기화

        return answer

불리언 값을 담을 리스트 answer을 만든 후

반복문으로 리스트를 순회하며 여분 사탕 개수를 더한다.

 

전체 원소 중 최대값과 같거나 크다면 최대값이므로 True를 

answer에 추가하고 초기화, 아니라면 False를 추가한 후 초기화한다.

모든 반복이 끝나면 각 원소의 불리언 값이 answer에 담긴다.