코딩테스트/Python

[LeetCode] 605. Can Place Flowers

swwho 2025. 3. 18. 10:41
728x90
반응형

🔗 Problem Link

https://leetcode.com/problems/can-place-flowers/description


❔Thinking

  • 0과 1로만 이루어진 flowerbed 배열이 주어질 때 주어진 꽃을 모두 심을 수 있다면 True, 아니라면 False를 반환한다.
  • 0인 곳에 꽃을 하나 심을 수 있고, 1이 인접하지 않아야 한다. (ex - [1,1], [0,1,1] 불가능)

💻Solution

1. 초기 풀이

def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:
    flowers = 0
    if flowerbed == [0] or (flowerbed == [1] and n == 0): return True
    elif flowerbed == [1]: return False

    for i in range(len(flowerbed)):
        if (i == 0 and flowerbed[i] == flowerbed[i+1] == 0):
            flowerbed[i] = 1
            flowers += 1
        elif (i == len(flowerbed)-1 and flowerbed[i] == flowerbed[i-1] == 0):
            flowerbed[i] = 1
            flowers += 1
        elif (flowerbed[i-1] == flowerbed[i] == flowerbed[i+1] == 0):
            flowerbed[i] = 1
            flowers += 1
    return True if flowers>=n else False

 

2. 개선된 풀이

def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:
    flowers, size = 0, len(flowerbed)
    
    for i in range(size):
        if flowerbed[i] == 0 and (i == 0 or flowerbed[i - 1] == 0) and (i == size - 1 or flowerbed[i + 1] == 0):
            flowerbed[i] = 1
            flowers += 1
            if flowers >= n:
                return True
    
    return flowers >= n

🗝️keypoint

  1. 현재를 기준으로 이전과 이후가 0이라면, 현재를 1로 바꿀 수 있다.
  2. 이전과 이후의 값을 확인할 경우, 배열의 크기 밖으로 벗어나는 확인은 에러가 발생한다.
  3. 현재가 0인지 확인하고, 인덱스가 양 끝인지 확인하고, 앞뒤가 0인지 확인하는 세 가지를 and로 묶어 확인할 수 있다.