605. 种花问题（Python）

题目

难度：★★☆☆☆
类型：数组

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

给定一个花坛（表示为一个数组包含0和1，其中0表示没种植花，1表示种植了花），和一个数 n 。能否在不打破种植规则的情况下种入 n 朵花？能则返回True，不能则返回False。

注意
数组内已种好的花不会违反种植规则。
输入的数组长度范围为 [1, 20000]。
n 是非负整数，且不会超过输入数组的大小。

示例

示例 1:
输入: flowerbed = [1,0,0,0,1], n = 1
输出: True

示例 2:
输入: flowerbed = [1,0,0,0,1], n = 2
输出: False

解答

这道题需理清逻辑，我们可以使用一个宽度为3的滑窗，这里通过指针i表示滑窗中心，则滑窗中包含3个元素[i-1, i, i+1]，指针i从第二个元素开始遍历到倒数第二个元素，在遍历的过程中，我们查看滑窗中是否所有位置都没有种花，如果没有，那么我们在滑窗中心栽一朵，否则继续遍历，如果手头没有剩下花了，说明可以栽下那么多，否则不能。这里需要注意的是，我们要考虑两端的情况，需要将数组做padding（左右各增加一个0）。

class Solution:
    def canPlaceFlowers(self, flowerbed, n):

        flowerbed = [0] + flowerbed + [0]                               # 左右各加一个空位置
        for i in range(1, len(flowerbed)-1):                            # 遍历出两端之外的位置
            if flowerbed[i-1] == flowerbed[i] == flowerbed[i+1] == 0:   # 遇到连续三个空位置
                flowerbed[i] += 1                                       # 放一盆花
                n -= 1                                                  # 花数减一
            if n <= 0:                                                  # 如果花用完了，返回True
                return True
        return False                                                    # 花没用完，返回False

如有疑问或建议，欢迎评论区留言~