Leetcode 605 种花问题

Leetcode 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

思路
直接贪心算法求解，一个一个遍历，左右都没花就可以种，可以将循环 i+2 , 如果 flowerbed[i-1] 和 flowerbed[i+1] 都为零，那么 flowerbed[i] 就可以种树；但是需要考虑到
1）如果开头第一个是0，第二个也是0，那么第一个就可以种树
2）奇数个数结尾，和偶数个数结尾，算的方法不一样。

最终代码

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        int len = flowerbed.size();
        if(len == 1)
        {
            if(flowerbed[0] == 1)
                return n == 0;
            else
                return true;
        }
        int k = 0;
        for(int i = 0; i < len; i += 2)
        {
            if(flowerbed[i] == 0)
            {
                if(i == 0 && flowerbed[i+1] == 0) //开头
                    k++;
                else if(i != 0 && i+1 != len && flowerbed[i-1] == 0 && flowerbed[i+1] == 0) //中间
                    k++;
                else if(i+1 == len && flowerbed[i-1] == 0)//结尾 需考虑元素的个数奇数
                    k++;
                else if(i+1 == len-1 && flowerbed[i+1] == 0)//结尾 需考虑元素的个数偶数
                    k++;
            }
        }
        return k >= n ? true:false;        
    }
};