Leetcode 1431 풀이(파이썬, 스위프트)

extraCandies를 캔디 배열 값에 더했을때 다른 수보다 작지 않을 수 있냐는 문제.

max함수로 최대값을 미리 저장해 두고 포문을 돌리며 비교

 

스위프트:

class Solution {
    func kidsWithCandies(_ candies: [Int], _ extraCandies: Int) -> [Bool] {
        var maxcan = candies.max()
        var a = [Bool]()
        for i in 0..<candies.count{
            if candies[i] + extraCandies >= maxcan!{
                a.append(true)
                
            }
            else{
                a.append(false)
            }
        }
        return a
    }
}

파이썬:

class Solution:
    def kidsWithCandies(self, candies: List[int], extraCandies: int) -> List[bool]:
        maxcand = max(candies)
        arr=[]
        for i in range(len(candies)):
            if candies[i]+extraCandies >= maxcand:
                arr.append(True)
            else:
                arr.append(False)
        return arr