LeetCode 605. 种花问题

示例 1:
输入: flowerbed = [1,0,0,0,1], n = 1
输出: True

示例 2:
输入: flowerbed = [1,0,0,0,1], n = 2
输出: False

注意:
数组内已种好的花不会违反种植规则。
输入的数组长度范围为 [1, 20000]。
n 是非负整数，且不会超过输入数组的大小。

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
    	if(n == 0)//不用种，肯定可以
    		return true;
        int i = 0, count = 0, end = flowerbed.size()-1;
        bool prevFlower = false;//前面没有花
        bool nextFlower = false;//后面有没有花
        while(i <= end)
        {
        	if(flowerbed[i] == 0)//当前可以种花
        	{
        		nextFlower = (i==end ? false : flowerbed[i+1]);
        		prevFlower = (i==0 ? false : flowerbed[i-1]);
        		if(!prevFlower && !nextFlower)
        		{
        			flowerbed[i] = 1;//种上一朵花
        			count++;//计数+1
        			if(count == n)
        				return true;
        		}
        	}
        	++i;
        }
        return false;
    }
};