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

提示：

1 <= flowerbed.length <= 2 * 104
flowerbed[i] 为 0 或 1
flowerbed 中不存在相邻的两朵花
0 <= n <= flowerbed.length

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

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

class Solution:
    def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:
        size = len(flowerbed)
        if size == 1:
            if flowerbed[0] and n:
                return False
            else:
                return True
        for i in range(size - 1):
            if (flowerbed[i] == 0 and flowerbed[i+1] == 0) and (i == 0 or i == size - 2 or flowerbed[i-1] == 0):
                flowerbed[i] = 1
                n = n - 1
        return n <= 0