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

题解

判断能否在不打破种植规则的情况下在花坛内种入 n 朵花，从贪心的角度考虑，应该在不打破种植规则的情况下种入尽可能多的花，然后判断可以种入的花的最多数量是否大于或等于 n。

代码

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        int i,count;
        int size=flowerbed.size();
        count=0;
        for(i=0;i<size;){
            if(size==1){
                if((flowerbed[0]==0&&n==1) || n==0)
                    return true;
                return false;
            }
//要考虑边界条件的不同
            if(i==0){
                if(flowerbed[i]==0&&flowerbed[i+1]==0){
                    count++;
                    i+=2;
                }else
                    i++;
            }else if(i==size-1){
                if(flowerbed[i]==0&&flowerbed[i-1]==0)
                    count++;
                    i++;
            }else if(i==size-1-1){
                if(flowerbed[i-1]==0&&flowerbed[i]==0&&flowerbed[i+1]==0){
                    count++;
                    i+=2;
                }else
                    i++;
            }else{
                if(flowerbed[i]==0&&flowerbed[i-1]==0&&flowerbed[i+1]==0){
                    count++;
                    i+=2;
                }else{
                    i++;
                }
            }
            if(count>=n)
                return true;
        }
        return false;
    }
};