题目: 假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,因为它们会争夺水源,两者都会死去。
给定一个整数数组 flowerbed 表示花坛,由若干 0 和 1 组成,其中 0 表示没有种植花,1 表示种植了花。另有一个数 n, 能否在不打破种植规则的情况下种入n 朵花?能则返回 True, 不能则返回 False
输入: flowerbed = [1,0,0,0,1], n = 2
输出: false
方法一:** 贪心算法**
class Solution {
public:
bool canPlaceFlowers(vector<int>& flowerbed, int n) {
int count = 0;
int m = flowerbed.size();
int prev = -1;
for (int i=0; i<m; i++){
if (flowerbed[i] == 1){
if(prev < 0){
count += i/2;
} else {
count += (i-prev-2)/2;
}
if(count >= n){
return true;
}
prev = i;
}
}
if(prev < 0){
count += (m+1)/2;
} else {
count += (m-prev-1)/2;
}
return count >= n;
}
};在这里插入代码片
复杂度分析
时间复杂度: O(m),其中 m 是数组 flowerbed 的长度。需要遍历数组一次。
空间复杂度: O(1)。额外使用的空间为常数。