力扣605(java&python)-种花问题（简单）

题目：

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

给你一个整数数组  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 * 10⁴
• flowerbed[i] 为 0 或 1
• flowerbed 中不存在相邻的两朵花
• 0 <= n <= flowerbed.length

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

解题思路：

就是当前位置的左右相邻位置都没有种花（为0），那么当前位置就为种花（设为1），能种花存在以下几种情况：

• 当前位置flowerbed[i] == 0，并且当前位置是数组的末尾：i = flowerbed.length - 1，并且它的前一个位置没有种花：flowerbed[i -1] == 0;
• 当前位置flowerbed[i] == 0，并且当前位置是数组第一个：i == 0，并且它的下一个位置没有种花：flowerbed[i + 1] == 0；
• 当前位置flowerbed[i] == 0，并且数组只有一个元素，i == 0并且 i = flowerbed.length - 1；
• 当前位置flowerbed[i] == 0，并且它的前一位置没有种花，flowerbed[i -1] == 0，并且它的后一位置也没有种花：flowerbed[i +1] == 0；

java代码：

class Solution {
    public boolean canPlaceFlowers(int[] flowerbed, int n) {
        int m = flowerbed.length;
        int count = 0;
        for(int i = 0; i < m; i++){
            if(flowerbed[i] == 0 && (i == m-1 || flowerbed[i+1] == 0) && (i == 0 || flowerbed[i-1] == 0)){
                flowerbed[i] = 1;
                count ++;
            }
        }
        return count >= n;
    }
}

python3代码：

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