每日一题-种花问题

---

题目

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/can-place-flowers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

示例

示例 1：

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

示例 2：

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

提示：

• 1 <= flowerbed.length <= 2 * 10⁴
• flowerbed[i] 为 0 或 1
• flowerbed 中不存在相邻的两朵花
• 0 <= n <= flowerbed.length

解题

我们给后面补0，解决后面遍历不到的问题。
前面遍历不到可以判断第一个元素和第二个元素都是0，则可以种一棵树，并将第一个元素直接改为1，不会影响后面的判断。

class Solution:
    def canPlaceFlowers(self, flowerbed, n):
        flowerbed.append(0)
        if flowerbed[0] == flowerbed[1] == 0:
            n -= 1
            flowerbed[0] = 1
        left = 1
        while left < len(flowerbed) - 1:
            if flowerbed[left - 1] == 0 and flowerbed[left + 1] == 0 and flowerbed[left] == 0:
                flowerbed[left] = 1
                left += 2
                n -= 1
            else:
                left += 1
        if n < 1:
            return True
        return False