题目
假设你有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花卉不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给定一个花坛(表示为一个数组包含0和1,其中0表示没种植花,1表示种植了花),和一个数 n 。能否在不打破种植规则的情况下种入 n 朵花?能则返回True,不能则返回False。
注意:
- 数组内已种好的花不会违反种植规则。
- 输入的数组长度范围为 [1, 20000]。
- n 是非负整数,且不会超过输入数组的大小。
思路
典型的贪心算法。
遍历一次数组,针对遍历的不同位置做不同的判断:
- 如果
i == 0
, 检查flowerbed[0]
和flowerbed[1]
是否为0- 若同时为0,则令
flowerbed[0] = 1, n--
;
- 若同时为0,则令
- 如果
i == len - 1
,检查flowerbed[len-1]
和flowerbed[len-2]
是否为0,- 若同时为0,则令
flowerbed[len-1] = 1, n--
;
- 若同时为0,则令
- 否则,检查
flowerbed[i-1],[i],[i+1]
是否为0,- 若同时为0,则
令flowerbed[i] = 1,n
--;
- 若同时为0,则
遍历结束后,检查n是否大于0,
- 大于0,返回
false
; - 小于等于0,返回
true
;
实现
Java
class Solution {
public boolean canPlaceFlowers(int[] flowerbed, int n) {
int len = flowerbed.length;
if (len == 0) return false;
if (len == 1) {
if (n == 0) return true;
else if (n == 1) return flowerbed[0] == 0;
else return false;
}
for (int i = 0; i < len; i++) {
if (i == 0) {
if (flowerbed[i] == 0 && flowerbed[i+1] == 0) {
flowerbed[i] = 1;
n--;
}
}else if (i == len - 1) {
if (flowerbed[i] == 0 && flowerbed[i-1] == 0) {
n--;
flowerbed[i] = 1;
}
}else {
if (flowerbed[i] == 0 && flowerbed[i-1] == 0 && flowerbed[i+1] == 0) {
n--;
flowerbed[i] = 1;
}
}
}
return n > 0 ? false : true;
}
}