LeetCode 605.种花问题

题目

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

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

注意:

1. 数组内已种好的花不会违反种植规则。
2. 输入的数组长度范围为 [1, 20000]。
3. n 是非负整数，且不会超过输入数组的大小。

思路

典型的贪心算法。

遍历一次数组，针对遍历的不同位置做不同的判断：

• 如果i == 0， 检查flowerbed[0]和flowerbed[1]是否为0
  • 若同时为0，则令flowerbed[0] = 1, n--;
• 如果i == len - 1，检查flowerbed[len-1]和flowerbed[len-2]是否为0，
  • 若同时为0，则令flowerbed[len-1] = 1, n--;
• 否则，检查flowerbed[i-1],[i],[i+1]是否为0，
  • 若同时为0，则令flowerbed[i] = 1，n--；

遍历结束后，检查n是否大于0,

• 大于0，返回false；
• 小于等于0，返回true；

实现

Java

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