leetcode 605. 种花问题——数组

605. 种花问题

题目来源：力扣（LeetCode）
https://leetcode-cn.com/problems/can-place-flowers/

解题思路：

原问题分解为子问题 ——》 求解子问题的最优解 ——》 合并子问题最优解得到原问题最优解。

本问题种花元素的前后都必须为0。

那么此时我们考虑当前元素的点，若当前元素为1，此地不能种花；若当前元素不为1，此时就需要判断该点前后连个元素是否都为0，若是则表示该点可以种花，将其置为1，若不是就向后一位元素继续判断。同样也采用一个计数标志，当计数的值不比n的值小，满足了条件，就可以直接返回 true。

当访问第一个元素时，当前点没有相邻的前一个点；当访问最后一个元素时，当前点没有相邻的最后一个点，此时需要单独在上述循环前后单独讨论，因为判断当前点能种花即会对当前点置1.

还有就是数组长度不超过2，此时上面的判断是不需要的。所以在开始需要单独讨论数组长度不超过2的情况。

当 n = 0 时，对所有数组都满足，所以可以直接返回 true，所以在进入数组前需要判断n的值。

代码：
代码有点累赘，主要说明思路，简化的代码可以看官方的解题代码。

class Solution {
    public boolean canPlaceFlowers(int[] flowerbed, int n) {
        int flag = 0;
        if(n == 0){
            return true;
        }
        if(flowerbed.length <=2){
            for(int i=0; i<flowerbed.length; i++){
                if(flowerbed[i]==1 && n>=1){
                    return false;
                }
            }
            flag++;
            if(flag >= n) 
                return true;
            else 
                return false;
        }
        if(flowerbed[0]==0 && flowerbed[1]==0){
            flowerbed[0] = 1;
            flag++;
            if(flag >= n) return true;
        }
        for(int i=1; i<flowerbed.length-1; i++){
            if(flowerbed[i]==0 && flowerbed[i-1]==0 && flowerbed[i+1]==0){
                flowerbed[i] = 1;
                flag++;
                if(flag >= n) return true;
            }
        }
        if(flowerbed[flowerbed.length-2]==0 && flowerbed[flowerbed.length-1]==0){
            flowerbed[flowerbed.length-1] = 1;
            flag++;
            if(flag >= n) return true;
        }
        return false;
    }
}

官方的简化解题代码：

public class Solution {
    public boolean canPlaceFlowers(int[] flowerbed, int n) {
        int i = 0, flag = 0;
        while (i < flowerbed.length) { // 包含数组长度不超过2的情况。
        	// 将前面代码讨论的情况都合在一起了，代码量大大减少。 
        	// (i == 0 || flowerbed[i - 1] == 0) 包含当前点含有前一个元素与不含前一个元素的情况。
        	// (i == flowerbed.length - 1 || flowerbed[i + 1] == 0) 包含当前点含有后一个元素与不含后一个元素的情况。
            if (flowerbed[i] == 0 && (i == 0 || flowerbed[i - 1] == 0) && (i == flowerbed.length - 1 || flowerbed[i + 1] == 0)) {
                flowerbed[i++] = 1;
                flag++;
            }
             if(flag>=n)
                return true;
            i++;
        }
        return false;
    }
}