种花问题（Python）

605. 种花问题

假设你有一个很长的花坛，一部分地块种植了花，另一部分却没有。

可是，花卉不能种植在相邻的地块上，它们会争夺水源，两者都会死去。

给定一个花坛（表示为一个数组包含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

def func(flowerbed, n):
    res = 0
    lenth = len(flowerbed)
    if lenth == 1:
        if flowerbed[0] == 0:
            res += 1
        return res>=n
    if lenth == 2:
        if flowerbed[0] == 0 and flowerbed[1] == 0:
            res += 1
        return res>=n
    for i in range(0, lenth-1):
        if i == 0 and flowerbed[0] == 0 and flowerbed[1] == 0:
            res += 1
            flowerbed[0] = 1
        if flowerbed[i-1] == 0 and flowerbed[i+1] == 0 and flowerbed[i] == 0:
            flowerbed[i] = 1
            res += 1
    if flowerbed[-1] == 0 and flowerbed[-2] == 0:
        res += 1
        flowerbed[-1] = 1
    return res>=n
print(func(flowerbed = [1], n = 2))