【LeetCode】种花问题

class Solution(object):
    def canPlaceFlowers(self, flowerbed, n):
        """
        :type flowerbed: List[int]
        :type n: int
        :rtype: bool
        """
        count = 0
        can_place = 0
        for index, i in enumerate(flowerbed):
            if index == 0 and i == 0:
                count = 2
                continue
            if index == (len(flowerbed) - 1) and i == 0:
                count += 1
            count = count + 1 if i == 0 else 0
            if count >= 3:
                can_place += 1
                count = 1
        if can_place >= n:
            return True
        elif len(flowerbed) == 1 and i==0:
            return True
        else:
            return False