605. Can Place Flowers（種花問題）

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. 數組內已種好的花不會違反種植規則。
2. 輸入的數組長度範圍爲 [1, 20000]。
3. n 是非負整數，且不會超過輸入數組的大小。

解法一

//時間複雜度O(n), 空間複雜度O(1)
class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        int len = flowerbed.size(), count = 0;
        for(int i = 0; i < len; i++) {
            if(!flowerbed[i] && (!i || !flowerbed[i - 1]) && (i == len - 1 || !flowerbed[i + 1])) {
                count++;
                i++;
            }
            if(count >= n) return true;
        }
        return false;
    }
};

思路：

從前往後遍歷，能種則種，同時計數，達到n時返回true；如果到結束都沒達到n，返回false。

2019/06/09 01:55