leetcode605. 种花问题

假设你有一个很长的花坛，一部分地块种植了花，另一部分却没有。可是，花卉不能种植在相邻的地块上，它们会争夺水源，两者都会死去。
给定一个花坛（表示为一个数组包含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的情况，然后再考虑如果数组头部有连续的两个0，那么可以在头部种一朵花，同理，如果尾部有连续的两个0，那么可以在尾部种一朵花，种完要将这个位置的0置为1，代表这个位置已经种花了，然后开始处理中间的情况，对中间的情况而言，如果存在连续的三个0，那么就可以在中间的那个位置上种花，种完置为1，这里的flag表示可以种植花的最大数量，最后得到的flag如果大于或等于n，即可以种植n朵花，反之则不行。有问题可以指出来，有更好的方法欢迎评论！

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        if(flowerbed.size()==1 && flowerbed[0]==0)
            return n<=1? true:false;
        if(flowerbed.size()==1 && flowerbed[0]==1)
            return n==0? true:false;
        int nCount = flowerbed.size();
        int flag = 0;
        if(flowerbed[0] == 0 && flowerbed[1] == 0)
        {
            flowerbed[0] = 1;
            flag++;
        }
        for(int i = 1;i<nCount-1;i++)
        {
            if(flowerbed[i-1]==0 && flowerbed[i]==0 && flowerbed[i+1]==0)
            {
                flowerbed[i] = 1;
                flag++;
            }
        }
        if(flowerbed[nCount-1]==0 && flowerbed[nCount-2]==0)
        {
            flag++;
        }
        if(flag >= n)
            return true;
        return false;
    }
};