假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给你一个整数数组 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
class Solution {
public:
bool canPlaceFlowers(vector<int>& flowerbed, int n) {
}
};
思路:
这题思路也很清晰,因为花不能相邻,所以能种花情形是:当前位置有花后面空三个位置再有花,我们的花才能种在中间;最大化情况就是尽量每朵花中间只空一格;考虑两端边缘也为空地时,花优先种在两端。结果就出来了。
class Solution {
public:
bool canPlaceFlowers(vector<int>& flowerbed, int n) {
int len=flowerbed.size();
int grownum=0;
if(n==0)
return true;
if(len==0)
return false;
if(len==1)
return flowerbed[0]==0 && n==1;
if(flowerbed[0]==0 && flowerbed[1]==0){
flowerbed[0]=1;
++grownum;
}
if(flowerbed[len-1]==0 && flowerbed[len-2]==0){
flowerbed[len-1]=1;
++grownum;
}
for(int i=1;i<len-1;++i){
if(flowerbed[i-1]==0 && flowerbed[i]==0 && flowerbed[i+1]==0){
flowerbed[i]=1;
++grownum;
}
}
if(grownum>=n)
return true;
return false;
}
};