力扣刷题👊【605. 种花问题】

刷题也是一种解压方式，当我们被工作或者生活压得喘不过气时，刷题能够让我们暂时忘却烦恼，沉浸在解题的乐趣中。而今天我们一起来看看力扣的第605题——种花问题。

题目背景

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

题目要求

给你一个整数数组 flowerbed，其中 0 表示该地块里没有花，1 表示该地块里种了花。另给你一个数 n ，请你种 n 朵花，并尽量使这 n 朵花距离最远。

示例

输入：flowerbed = [1,0,0,0,1], n = 1
输出：true

输入：flowerbed = [1,0,0,0,0,1], n = 2
输出：false

解题思路

本题的解题思路很简单，我们可以从花坛的左端开始遍历，每当我们遇到一个空地块时，我们就可以将一朵花种在该地块上。但是，我们需要确保这朵花与之前种植的花之间至少有一个空地块，以避免它们争夺水源。

算法步骤

1. 从花坛的左端开始遍历。
2. 每当我们遇到一个空地块时，我们就可以将一朵花种在该地块上。
3. 但是，我们需要确保这朵花与之前种植的花之间至少有一个空地块，以避免它们争夺水源。
4. 如果我们遍历完整个花坛，并且成功地种下了 n 朵花，那么我们就返回 true。
5. 否则，我们就返回 false。

代码实现

def canPlaceFlowers(flowerbed, n):
    # 如果花坛的长度为 0，那么我们无法种下任何花朵。
    if len(flowerbed) == 0:
        return False

    # 如果花坛的长度为 1，并且花坛的第一个地块是空的，那么我们可以种下一朵花。
    if len(flowerbed) == 1 and flowerbed[0] == 0:
        return True

    # 遍历花坛，每当我们遇到一个空地块时，我们就将一朵花种在该地块上。
    for i in range(1, len(flowerbed) - 1):
        # 如果当前地块是空的，并且前一个地块和后一个地块都是空的，那么我们就将一朵花种在当前地块上。
        if flowerbed[i] == 0 and flowerbed[i - 1] == 0 and flowerbed[i + 1] == 0:
            flowerbed[i] = 1
            n -= 1

    # 如果我们遍历完整个花坛，并且成功地种下了 n 朵花，那么我们就返回 true。
    if n == 0:
        return True

    # 否则，我们就返回 false。
    else:
        return False


# 测试用例
flowerbed1 = [1, 0, 0, 0, 1]
n1 = 1
flowerbed2 = [1, 0, 0, 0, 0, 1]
n2 = 2

# 调用函数并打印结果
print(canPlaceFlowers(flowerbed1, n1))  # 输出：true
print(canPlaceFlowers(flowerbed2, n2))  # 输出：false