贪心605. 种花问题

Posted 2021-09-23 7TribeZ

605. 种花问题

难度简单381

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

给你一个整数数组  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

解析：

1.贪心

【思路】
从左向右遍历花坛，在可以种花的地方就种一朵，能种就种（因为在任一种花时候，不种都不会得到更优解），就是一种贪心的思想
这里可以种花的条件是：

自己为空
左边为空 或者 自己是最左
右边为空 或者 自己是最右
最后判断n朵花是否有剩余，为了效率起见，可以在种花的过程中做判断，一旦花被种完就返回true

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        int total=0;
        int h=flowerbed.size();
        for (int i=0;i<h;i++){
            //顺序不能乱，先判断是不是i==0或者i==h-1 是则不执行||后面的内容
             if(flowerbed[i] == 0 && (i == 0 || flowerbed[i-1] == 0) && (i == h-1 || flowerbed[i+1] == 0)) {
                total++;
                flowerbed[i]=1;
            }
        }
        if(total>=n) return true;
        else return false; 
    }
};

2.前后添0，统一判断条件

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        flowerbed.insert(flowerbed.begin(),0);
        flowerbed.insert(flowerbed.end(),0);

        int h=flowerbed.size();
        int total=0;
        for (int i=0;i<h-2;i++)
        {
           
            if(flowerbed[i]==0 &&flowerbed[i+1]==0&&flowerbed[i+2]==0){
                flowerbed[i+1]=1;
                total++;
            }
     
        }
        if (total>=n) return true;
        else    return false;
    }
};