LeetCode 种花问题

# !/usr/bin/env python3
# -*- coding: UTF-8 -*-
"""
@author  : v_jiaohaicheng@baidu.com
@des     : 假设有一个很长的花坛，一部分地块种植了花，另一部分却没有。可是，花不能种植在相邻的地块上，它们会争夺水源，两者都会死去。

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

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


"""


class Solution:
    def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:
        if n == 0:
            return True
        # 记录成功种花数量
        target = 0
        # 前后各插入一个0，空位，只需考虑连续三个0在一起的情况+1即可
        flowerbed.insert(0, 0)
        flowerbed.append(0)
        # 从第一个位置开始，因为0位置一直都是0
        i = 1
        # 限制边界
        while i < len(flowerbed) - 1:
            if flowerbed[i] == 0:
                if flowerbed[i - 1] == 0 and flowerbed[i + 1] == 0:
                    # 种花一颗
                    target += 1
                    # 这三个相邻0跳过
                    i += 2
                    # 减少循环次数，条件成立，直接符合题意，终止循环，不符合条件继续循环
                    if target >= n:
                        return True

                else:
                    i += 1
            # 继续循环
            else:
                i += 1
        # 不符合条件
        return False