605. 種花問題
阿新 • • 發佈:2019-01-31
假設你有一個很長的花壇,一部分地塊種植了花,另一部分卻沒有。可是,花卉不能種植在相鄰的地塊上,它們會爭奪水源,兩者都會死去。
給定一個花壇(表示為一個數組包含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, 20000]。
n 是非負整數,且不會超過輸入陣列的大小。
要想根據連續0的個數來計算能種多少花,需要考慮邊界的問題,為了解決邊界問題,可以採取補0的方法,如果第一個元素為0,就在最前面再補一個0,如果最後一個元素為0,就在最後面再補一個0
class Solution {
public:
bool canPlaceFlowers(vector<int>& flowerbed, int n) {
if(flowerbed.empty()){
return false;
}
if(flowerbed[0]==0){
flowerbed.insert(flowerbed.begin(),0 );
}
if(flowerbed.back()==0){
flowerbed.push_back(0);
}
int sum = 0,cnt = 0;
for(int i=0;i<=flowerbed.size();i++){
if(i<flowerbed.size()&&flowerbed[i]==0){
cnt++;
}else{
sum += (cnt-1)/2 ;
cnt = 0;
}
}
return sum >= n;
}
};