力扣 leetcode 605. 种花问题

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

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

Example1:

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

Excxample2:

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

思路：
首先整体思路是计算最大种花量为ans
与需要种花量n进行比较
若ans大于或等于n则能种n朵返回True
若ans小于n则不能种n朵花返回False

之后进行分组讨论

一、对于两侧的两个花坛（索引为0， -1的花坛）：
只要两侧的两个花坛旁边的花坛为0
两侧的花坛就可以种花

二、对于中间的花坛（除了索引为0， -1， 1， -2）的花坛：
只有周围的两个花坛均为0时
中间的花坛可以种花
（由于遍历的花坛已经种花，则下一个花坛无法种花，直接跳过下一个花坛）

三、对于索引为1， -2的两个花坛
这两个花坛若周围均为0，则将花种在两侧或这两个花坛结果是一样的
例如

若开始的序列为0，0，0，1
则将话种在索引0，或者索引1都是只能种一朵

Code：

class Solution:
    def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:
        ans = 0
        if len(flowerbed) == 1:
            if flowerbed[0] == 0 and n <= 1:
                return True
            elif n == 0:
                return True
            else:
                return False
            

        if flowerbed[0] == 0 and flowerbed[1] == 0:
            ans += 1
            flowerbed[0] = 1
        if flowerbed[-2] == 0 and flowerbed[-1] == 0:
            ans += 1
            flowerbed[-1] = 1

        i = 2
        while i < len(flowerbed) - 2:
            if flowerbed[i - 1] == 0 and  flowerbed[i + 1] == 0 and flowerbed[i] != 1:
                flowerbed[i] = 1
                ans += 1
                i += 1
            i += 1
        
        if ans >= n:
            return True
        else:
            return False

运行效果：