LeetCode605种花问题

问题描述

  假设有一个很长的花坛，一部分地块种植了花，另一部分却没有。可是，花不能种植在相邻的地块上，它们会争夺水源，两者都会死去。给你一个整数数组 flowerbed 表示花坛，由若干 0 和 1 组成，其中 0 表示没种植花，1 表示种植了花。另有一个数 n ，能否在不打破种植规则的情况下种入 n 朵花？能则返回 true ，不能则返回 false 。

解析

  这题其实没什么难度，暴力方法的复杂度也就是O(n)，从头遍历，如果周围两个都是空位则说明可以种花。官方解答中定义了三个额外变量，实际上是不需要的，例如count，官方是每种一朵花count++，然后和n去比较，实际上不用引入额外的count空间，每种一朵花n–，然后将n和0比较即可。
  然后就是关于要不要把第一个和最后一个位置的判断写到循环里，如果循环次数非常大的话还是推荐写出来，这样每次循环就能少判断一个条件，会节省一点时间，这里因为数据量并不大所以时间差不多。每次循环后判断是否结束也是一样，根据数据可能是负优化（例如对于某些数据，这个if从未被触发，会引入额外的判断开销）。

class Solution {
    public boolean canPlaceFlowers(int[] flowerbed, int n) {
        if (flowerbed.length == 1) {
            if (flowerbed[0] == 0) {
                return n <= 1;
            }
            else {
                return n == 0;
            }
        }

        if (flowerbed[0] == 0 && flowerbed[1] == 0){
            n--;
            flowerbed[0] = 1;
        }
        for(int i = 1; i < flowerbed.length - 1; i++){
            if(flowerbed[i] == 0 && flowerbed[i - 1] == 0 && flowerbed[i + 1] == 0){
                n--;
                flowerbed[i] = 1;
            }
            if(n == 0){
                return true;
            }
        }
        if (flowerbed[flowerbed.length-1] == 0 && flowerbed[flowerbed.length-2] == 0){
            n--;
        }
        return n <= 0;
    }
}