【力扣练习记录】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

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        int cnt=n;
        if(n==0)return true;
        if(n>flowerbed.size())return false;
        if(flowerbed.size()==1){
             if(flowerbed[0]==1)return false;
             if(flowerbed[0]==0&&n==1)return true;
        }

           for(int i=0;i<flowerbed.size();i++){
               if(cnt==0)break;
            if(i-1>=0&&i+1<=flowerbed.size()-1){
                 if(flowerbed[i]==0&&flowerbed[i+1]==0&&flowerbed[i-1]==0){
                   flowerbed[i]=1;
                   //cout<<i<<endl;
                   cnt--;
               }
            }

           else if(i==0){
              //第一个数
               if(flowerbed[i]==0&&flowerbed[i+1]==0){
                   flowerbed[i]=1;
                   cnt--;
               } 
          }
            else if(i==flowerbed.size()-1){
               if(flowerbed[i]==0&&flowerbed[i-1]==0){
                   flowerbed[i]=1;
                   cnt--;
               } 
            }

           }

           if(cnt==0)return true;
           else return false;
    }
};