LeetCode刷题日记-数组-面试题 605. 种花问题

题目描述

题目链接：https://leetcode-cn.com/problems/can-place-flowers/

解题思路

看到这题，第一时间想到连续三个元素都是为零的话，就可以在中间位置种上种上一棵花。但是考虑到了边界问题，比如[0 0 0]，如果按照上面的思路，种的结果结束[0 1 0 ]。而实际上可以种两颗[1 0 1]。
上面的思想出发点没错，但有一个重要的缺点就是：遍历过程中，边界元素没有作为中间元素，看了看评论区，网友们的确实聪明绝顶！边界填充0的话，这样遍历过程每个元素都可以作为中间元素，非常巧妙！犹如卷积一样，边界填充零。
还有一点就是，最开始想定义一个变量记录种下的数量，但根据题目意思，只要种满n颗就可以满足条件了。直接n -= 1就行，而不需要额外的变量来进行自增运算。

解题代码如下：

class Solution:
    def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:
        flowerbed = [0] + flowerbed + [0]   # 边界填充零
        for i in range(1, len(flowerbed)-1):  # 注意遍历范围，填充之后，首尾都是零
            if (flowerbed[i-1]==0 and flowerbed[i]==0 and flowerbed[i+1]==0):
                flowerbed[i] = 1
                n -= 1 
        return n<=0