LEETCODE | PYTHON | 605 | 种花问题

LEETCODE | PYTHON | 605 | 种花问题

1. 题目

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

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

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

2. 代码

class Solution:
    def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:
        #初始化
        l = len(flowerbed)
        res = 0

        #两头插入0补充虚拟的位置方便遍历
        flowerbed.append(0)
        flowerbed.insert(0,0)

        #遍历flowerbed判断可插入花的位置
        for i in range(1,l+1):
            #如果左右都为0且当前位置为0则可以种花
            if flowerbed[i-1] + flowerbed[i+1] == 0 and flowerbed[i] == 0:
                res = res + 1
                #更新当前位置：已种花
                flowerbed[i] = 1 
        
        return True if res>=n else False

3. 思路

(1) 为了方便后续遍历，在flowerbed两头增加0，即虚拟空位
(2) 可以种花的条件为：左右两边都为空
(3) 若达到条件，计数器res+=1，且需更新flowerbed在节点i的状态