力扣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代码：

 1 class Solution {
 2     public boolean canPlaceFlowers(int[] flowerbed, int n) {
 3         int m = flowerbed.length;
 4         int count = 0;
 5         for(int i = 0; i < m; i++){
 6             if(flowerbed[i] == 0 && (i == m-1 || flowerbed[i+1] == 0) && (i == 0 || flowerbed[i-1] == 0)){
 7                 flowerbed[i] = 1;
 8                 count ++;
 9             }
10         }
11         return count >= n;
12     }
13 }

python3代码：

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