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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/can-place-flowers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
private:
    int count=0;
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        if(flowerbed.empty()){
            return false;
        }
        if(n>flowerbed.size()){
            return false;
        }
        if(flowerbed.size()==1){
            if(flowerbed[0]==0||n==0){
                return true;
            }
            return false;
        }
        if(flowerbed[0]==0&&flowerbed[1]==0){
            count++;
            flowerbed[0]=1;
        }
        int n1 = flowerbed.size();
        if(flowerbed[n1-1]==0&&flowerbed[n1-2]==0){
            count++;
            flowerbed[n1-1]=1;
        }
        for(int i=1;i<flowerbed.size()-1;++i){
            if(flowerbed[i-1]==1||flowerbed[i+1]==1||flowerbed[i]==1){
                continue;
            }else{
                count++;
                flowerbed[i]=1;
            }
        }
        //cout<<count<<endl;
        if(n>count){
            return false;
        }
        return true;
    }
};