LeetCode 605. 种花问题

1. 题目

假设你有一个很长的花坛，一部分地块种植了花，另一部分却没有。可是，花卉不能种植在相邻的地块上，它们会争夺水源，两者都会死去。

给定一个花坛（表示为一个数组包含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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

• 题目意思是，一个为0的位子，前后都是0才可以种植花朵
• 用两个bool变量表示当前位置i的前后一个位置有没有花，注意边界处理

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;
    }
};