605.种花问题[easy]

原题和解

https://leetcode-cn.com/problems/can-place-flowers/

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        int size_ = flowerbed.size();
        int total = 0, i = 1;
        if(size_ == 1 && flowerbed[0] == 0){
            total++;
            return total >= n;
        }
        if (flowerbed[0] == 0 && flowerbed[1] == 0){
            flowerbed[0] = 1;
            total++;
        }
        while(i < size_ - 1 && total < n){
            if (flowerbed[i - 1] == 0 && flowerbed[i + 1] == 0 && flowerbed[i] == 0){
                flowerbed[i] = 1;
                total++;
                i += 2;
            }
            else i++;
        }
        if (flowerbed[size_-1] == 0 && flowerbed[size_ - 2] == 0){
            total++;
        }
        cout<<total<<endl;
        return total >= n;
    }
};

思路

思路基本很简单，从左往右判断有三个连着的0，种了往下就行，注意处理下两边的特殊情况。

别的解法

看到一个结构更清晰简单的解法

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        // 每次跳两格
         for (int i = 0; i < flowerbed.size(); i += 2) {
             // 如果当前为空地
            if (flowerbed[i] == 0) {
                // 如果是最后一格或者下一格为空
                if (i == flowerbed.size() - 1 || flowerbed[i + 1] == 0) {
                    n--;
                } else {
                    i++;
                }
            }
        }
        return n <= 0;
    }
};

作者：heroding
链接：https://leetcode-cn.com/problems/can-place-flowers/solution/qi-si-miao-jie-by-heroding-h7m0/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

这个解法，在if里判断了是不是最后一个格子，所以不会出现越界，对于可能会越界的情况可以这样处理。