LeetCode算法题（数组相关）（十四）——种花问题

问题：

假设你有一个很长的花坛，一部分地块种植了花，另一部分却没有。可是，花卉不能种植在相邻的地块上，它们会争夺水源，两者都会死去。

给定一个花坛（表示为一个数组包含0和1，其中0表示没种植花，1表示种植了花），和一个数 n 。能否在不打破种植规则的情况下种入 n 朵花？能则返回True，不能则返回False。

输入: flowerbed = [1,0,0,0,1], n = 1
输出: True

输入: flowerbed = [1,0,0,0,1], n = 2
输出: False

算法：

因为只用了一个循环，所以可能效率高一点。我的算法是将给定的数组先给它种花，如下图所示：

实现：

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