第四天——贪心算法——种花

1. 题目

 

有一个花坛，其中0 表示该位置是空的，可以种花。1 表示该位置已经有花，不能种花。

规则：新种的花不能种在相邻的位置（即如果某个位置已经种了花，它的左右两个相邻位置不能再种花）。给定 花坛flowerbed 数组和需要新种的花的数量 n，判断是否可以成功种下 n 朵新花。

2. 示例

输入：

flowerbed = [1,0,0,0,1], n = 1

输出：True

解释：可以在中间位置种 1 朵花而不违反规则。

3. 解题思路

核心思想: 贪心算法（尝试在可以种花的位置立即种花，以尽早满足 n 的要求）。

1. 遍历数组：检查每一个位置是否可以种花。
2. 判断条件：
   • 当前位置是 0（空）。
   • 当前位置的左边是 0（或者是边界，即左边没有花）。
   • 当前位置的右边是 0（或者是边界，即右边没有花）。
3. 如果可以种花：
   • 把该位置设为 1（标记为已种）。
   • 计数器 count += 1（记录已种的花数）。
4. 终止条件：
   • 当 count >= n 时，可以直接返回 True（表示已经可以种 n 朵花了）。
   • 遍历结束后如果 count >= n，返回 True，否则返回 False。

4. 完整代码

def canPlaceFlowers(flowerbed, n):
    count = 0  # 记录可以种的花的数量
    length = len(flowerbed)
    for i in range(length):
        if flowerbed[i] == 0:  # 当前位置是空的
            # 检查左边（如果是第一个位置则左边视为空）
            left_empty = (i == 0) or (flowerbed[i-1] == 0)
            # 检查右边（如果是最后一个位置则右边视为空）
            right_empty = (i == length - 1) or (flowerbed[i+1] == 0)
            if left_empty and right_empty:
                flowerbed[i] = 1  # 种花
                count += 1
                if count >= n:  # 提前终止
                    return True
    return count >= n
print(canPlaceFlowers([1,0,0,0,1], 1))