力扣 605、种花问题

力扣 605、种花问题

假设你有一个很长的花坛，一部分地块种植了花，另一部分却没有。可是，花卉不能种植在相邻的地块上，它们会争夺水源，两者都会死去。
给定一个花坛（表示为一个数组包含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, 20000]。
n 是非负整数，且不会超过输入数组的大小。

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

这个题的意思说白了就是插入1之后相邻两数字不能相同，且我们要计算插入1的个数。
因此应该最先想到的方法就是连续三个数为0则满足条件，不瞒你说，我最先想到的也是这。但是这样写会存在边界问题，即：

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        int count = 0;//计数种n朵花
        for (auto i = flowerbed.begin()+1; i != flowerbed.end(); ++i){
            if((*i) == 0 && (*(i-1)) == 0 && (*(i+1)) == 0){
                (*i) = 1;
                count++;
            }
        }
        return n <= count;
    }
};

就是当第一个元素是0时，你的前一个元素是不存在的，也就无法判断。那么也会有人说那你从第二个元素开始判断啊，是的，但是还有一个问题，如果出现001的情况，第一位可以插入1满足条件，但是也会被判断为不符合，并且容器末尾100也会出现类似情况。因此一个好的解决方案就是在容器的前后两端先插入一个0，然后从第二个元素（原容器的第一个元素）开始遍历，即

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        int count = 0;//计数种n朵花
        flowerbed.insert(flowerbed.begin(),0);
        flowerbed.insert(flowerbed.end(),0);
        for (auto i = flowerbed.begin()+1; i != flowerbed.end()-1; ++i){
            if((*i) == 0 && (*(i-1)) == 0 && (*(i+1)) == 0){
                (*i) = 1;
                count++;
            }
        }
        return n <= count;
    }
};

在这里一定要注意开始遍历的位置是begin+1；退出位置为end-1，（end指向末尾元素的下一个位置）

此外，还有一种算法，从头开始遍历容器：

• 如果元素是0，先判断该元素是不是首元素，如果是首元素，那么就判断下一个元素是否为0，若为0，则该元素便可置1（种花 ），否则继续遍历。即（i == flowerbed.begin() && ((i+1)）== 0）；
  如果该元素为0且位于末尾，则判断前一个元素是否为0，为0则种花。即（i == flowerbed.end()-1 && ((i-1)) == 0）
• 如果该元素位于中间，那么和上一种算法一样，判断是否为连续三个0.
  代码如下：

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        int count = 0;//计数种n朵花
        for (auto i = flowerbed.begin(); i != flowerbed.end(); ++i){
            if((*i) == 0 && (i == flowerbed.begin() || (*(i-1)) == 0) && (i == flowerbed.end()-1 || (*(i+1)) == 0)){
                (*i) = 1;
                count++;
            }
        }
        return n <= count;
    }
};