605. 种花问题 2021vivo提前批笔试题

假设你有一个很长的花坛，一部分地块种植了花，另一部分却没有。
可是，花卉不能种植在相邻的地块上，它们会争夺水源，两者都会死去。
给定一个花坛（表示为一个数组包含0和1，其中0表示没种植花，1表示种植了花），和一个数 n 。
能否在不打破种植规则的情况下种入 n 朵花？能则返回True，不能则返回False。
示例 1:
输入: flowerbed = [1,0,0,0,1], n = 1
输出: True
示例 2:
输入: flowerbed = [1,0,0,0,1], n = 2
输出: False

解题思路

每种一个花，他的左右两边必须为0，它本身也要为0才可以种，所以条件是当连续三个数都为0时，可以在中间那个0上种一朵花，并且将其置为1。初略一想似乎还要考虑边界情况，不过其实可以在左右边界各添加一个0，即可解决边界问题。

class Solution(object):
    def canPlaceFlowers(self, flowerbed, n):
        """
        :type flowerbed: List[int]
        :type n: int
        :rtype: bool
        遍历一次
        """
        max_flower = 0 
        # 在左右边界各添加一个0，即可解决边界问题
        flowerbed = [0] + flowerbed + [0]
        for i in range(1, len(flowerbed)-1):
            if flowerbed[i]==0 and flowerbed[i-1]==0 and flowerbed[i+1]==0:
                max_flower += 1 
                flowerbed[i] = 1 
        return max_flower>=n