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 * 104
• flowerbed[i] 为 0 或 1
• flowerbed 中不存在相邻的两朵花
• 0 <= n <= flowerbed.length

题解：

/**
 * @param {number[]} flowerbed
 * @param {number} n
 * @return {boolean}
 */
// 方法一：扫描
var canPlaceFlowers = function(flowerbed, n) {
    if (n === 0) return true
    for (let i = 0; i < flowerbed.length; i++){
        let pre = flowerbed[i - 1] !== 1
        let cur = flowerbed[i] !== 1
        let nex =  flowerbed[i + 1] !== 1
        // 当 pre && cur && nex 满足条件时，判断可种植花朵数量
        // 当满足条件减少一朵后，不在有花朵就满足条件
        // 当满足条件减少一朵后，还有花朵，当前位置移动一位继续判断
        if (pre && cur && nex) {
            if (--n === 0) return true
            i++
        }
    }
    return false
};
// 方法二：贪心
var canPlaceFlowers = function(flowerbed, n) {
    let count = 0;
    const m = flowerbed.length;
    let prev = -1;
    for (let i = 0; i < m; i++) {
        if (flowerbed[i] === 1) {
            if (prev < 0) {
                count += Math.floor(i / 2);
            } else {
                count += Math.floor((i - prev - 2) / 2);
            }
            if (count >= n) {
                return true;
            }
            prev = i;
        }
    }
    if (prev < 0) {
        count += (m + 1) / 2;
    } else {
        count += (m - prev - 1) / 2;
    }
    return count >= n;
};

来源：力扣（LeetCode）

链接：leetcode.cn/problems/ca…

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。