1. 程式人生 > >605. 種花問題

605. 種花問題

假設你有一個很長的花壇,一部分地塊種植了花,另一部分卻沒有。可是,花卉不能種植在相鄰的地塊上,它們會爭奪水源,兩者都會死去。

給定一個花壇(表示為一個數組包含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; } };