JavaScript实现种植花朵

 

1 题目描述

https://leetcode-cn.com/problems/can-place-flowers/

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

给定一个花坛（表示为一个数组包含0和1，其中0表示没种植花，1表示种植了花），和一个数 n 。能否在不打破种植规则的情况下种入 n 朵花？能则返回True，不能则返回False

2 输入输出

示例 1:

输入: flowerbed = [1,0,0,0,1], n = 1

输出: True

示例 2:

输入: flowerbed = [1,0,0,0,1], n = 2

输出: False

注意:

数组内已种好的花不会违反种植规则。

输入的数组长度范围为 [1, 20000]。

n 是非负整数，且不会超过输入数组的大小。

3 解答：贪心算法

我们从左到右扫描数组 flowerbed，如果数组中有一个 0，并且这个 0 的左右两侧都是 0，那么我们就可以在这个位置种花，即将这个位置的 0 修改成 1，并将计数器 count 增加 1。对于数组的第一个和最后一个位置，我们只需要考虑一侧是否为 0。

在扫描结束之后，我们将 count 与 n 进行比较。如果 count >= n，那么返回 True，否则返回 False。（这里在for循环中对result和n的大小进行比较）

4代码

/*种花问题
*使用if...else if语句
 * @param {number[]} flowerbed
 * @param {number} n
 * @return {boolean}
 */
let canPlaceFlowers = function(flowerbed, n) {
    if(flowerbed.length <= 0){
        return false;
    }
    let result = 0;//result表示最多能种的花数
    if(flowerbed.length ===1 && flowerbed[0] === 0){ //长度为1时的判断
        result = 1;
    }
    for(let i = 0; i < flowerbed.length; i++){  //长度大于=2时
        if(i === 0){  //对于第一个元素
            if(flowerbed[i] === 0 && flowerbed[i + 1] === 0){
                result += 1;
                flowerbed[i] = 1;
            }
        }else if(i === flowerbed.length - 1){  //最后一个元素
            if(flowerbed[i] !== 1 && flowerbed[i - 1] === 0){
                result += 1;
                flowerbed[i] = 1;
            }
        }else{  //中间元素
            if(flowerbed[i] !== 1 && flowerbed[i - 1] === 0 && flowerbed[i + 1] === 0){
                result += 1;
                flowerbed[i] = 1;
            }
        }
    }
    return result >= n;
};

console.log(canPlaceFlowers([1,0,0,0,1,0,0], 2));

 

/*种花问题
*优化判断条件
 * @param {number[]} flowerbed
 * @param {number} n
 * @return {boolean}
 */
let canPlaceFlowers = function(flowerbed, n) {
    if(flowerbed.length <= 0){
        return false;
    }
    let result = 0;//result表示最多能种的花数
    //if(flowerbed.length ===1 && flowerbed[0] === 0){ //长度为1时的判断
    //    result = 1;
    //}
    for(let i = 0; i < flowerbed.length; i++){  //长度大于等于1时
        /*首先，先判断这个数是不是为0，如果是0；
        * 再判断是不是第一个元素，如果不是只需要判断前面那个是不是为0，如果是0；
        * 再判断是不是最后一个，如果不是，需要判断后面一个元素是不是为0
        *
        * 注意：当长度为1时，该元素是第一个元素也是最后一个元素*/
        if(flowerbed[i] === 0 && (i === 0 || flowerbed[i - 1] === 0) &&
            (i === flowerbed.length - 1 || flowerbed[i + 1] === 0)){//当长度为1时，i === flowerbed.length - 1为true
            result += 1;
            flowerbed[i] = 1;
        }
    }
    return result >= n;
};

console.log(canPlaceFlowers([1,0,0,0,1,0,0], 2));

 

参考：GitHub、力扣

百里于2020年5月4日

如果有错，请您指出！如有侵权，请联系我删除！