LeetCode 605. 种花问题

605. 种花问题

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

给你一个整数数组 flowerbed 表示花坛，由若干 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

提示：

• 1 <= flowerbed.length <= 2 * 10^4
• flowerbed[i] 为 0 或 1
• flowerbed 中不存在相邻的两朵花
• 0 <= n <= flowerbed.length

解法1：模拟

计算花坛中可以种入的花的最多数量，判断是否大于或等于 n 即可。

Java版：

class Solution {
    public boolean canPlaceFlowers(int[] flowerbed, int n) {
        int m = flowerbed.length;
        int cnt = 0;
        int i = 0;
        while (i < m) {
            if (flowerbed[i] == 0) {
                if ((i == 0 || flowerbed[i - 1] == 0) && (i == m - 1 || flowerbed[i + 1] == 0)) {
                    cnt++;
                    if (cnt == n) {
                        return true;
                    }
                    flowerbed[i] = 1;
                    i++;
                }
            }
            i++;
        }
        return cnt >= n;
    }
}

Python3版：

class Solution:
    def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:
        cnt = 0
        m = len(flowerbed) 
        i = 0
        while i < m:
            if flowerbed[i] == 0:
                if (i == 0 or flowerbed[i - 1] == 0) and (i == m - 1 or flowerbed[i + 1] == 0):
                    cnt += 1
                    if cnt == n:
                        return True
                    flowerbed[i] = 1
                    i += 1
            i += 1
        return cnt >= n

复杂度分析

• 时间复杂度：O(m)，其中 m 是数组 flowerbed 的长度。需要遍历数组一次。

• 空间复杂度：O(1)。额外使用的空间为常数。