力扣 leetcode 605. 种花问题 (python)

Topic

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

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

Example_1:

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

Exxample_2:

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

Solution：

首先整体思路是计算最大种花量为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