【力扣】605.种花问题--Python实现

【题目描述】
假设有一个很长的花坛，一部分地块种植了花，另一部分却没有。可是，花不能种植在相邻的地块上，它们会争夺水源，两者都会死去。

给你一个整数数组 flowerbed 表示花坛，由若干 0 和 1 组成，其中 0 表示没种植花，1 表示种植了花。另有一个数 n ，能否在不打破种植规则的情况下种入 n 朵花？能则返回 true ，不能则返回 false。

示例 1：
输入：flowerbed = [1,0,0,0,1], n = 1
输出：true

示例 2：
输入：flowerbed = [1,0,0,0,1], n = 2
输出：false

提示：
1 <= flowerbed.length <= 2 * 104
flowerbed[i] 为 0 或 1
flowerbed 中不存在相邻的两朵花
0 <= n <= flowerbed.length

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/can-place-flowers

【解题思路】
贪心法，有三种可能性：
0
00100
10001
分别对应着：
len(flowerber)==0
为首或者为尾且相邻的元素为0
为中间元素且左右都为0

用Python实现的代码如下：

class Solution(object):
    def canPlaceFlowers(self, flowerbed, n):
        """
        :type flowerbed: List[int]
        :type n: int
        :rtype: bool
        """
        for i in range(len(flowerbed)):
            if n==0:
                return True
            if flowerbed[i]==0:
                if len(flowerbed)==1:
                    n -= 1
                elif i==0 and flowerbed[i+1]==0:
                    n -= 1
                    flowerbed[i]=1
                elif i==len(flowerbed)-1 and flowerbed[i-1]==0:
                    n -= 1
                    flowerbed[i]=1
                elif 0<i<len(flowerbed)-1 and flowerbed[i-1]==0 and flowerbed[i+1]==0:
                    n -= 1
                    flowerbed[i]=1
        if n==0:
            return True
        else:
            return False