刷题之旅从数组类型的题目开始。第二十七道题目是种花问题。

image

题目描述

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

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

示例 1:

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

示例 2:

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

注意:

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

解题思路

此题思路比较清晰，其实连续三个0的时候就可以种一朵花。即i-1,i,i+1这三个地方元素都是0才行，不过判断完之后需要将i位置元素置为1，否则会造成后续程序的错误判断。需要考虑下边界，尤其是i+1不要越界。

看到一个优化的思路，就是两边各加一个空花盆避免越界，简化代码。具体思路见代码注释。

提交代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
class Solution {
    public boolean canPlaceFlowers(int[] flowerbed, int n) {
        //初始化计数
        int count = 0;
        //n=0表示不种花，那么必定符合条件，返回true
        if(n == 0){
            return true;
        }
        //考虑只有一个花盆的情况
        if(flowerbed.length == 1){
            if(flowerbed[0] == 0 && n <= 1){
                return true;
            }
            return false;
        }
        //考虑前两个花盆都是0的情况，判断了之后，第一个花盆置为1，否则会影响后续的判断
        if(flowerbed[0] == 0 && flowerbed[1] == 0){
            flowerbed[0] = 1;
            count++;
        }
        //考虑通用情况，只要i处可以放，就计数加一，并且将i处置为1，将这个花盆预占住
        for(int i=1;i<flowerbed.length;i++){
            if(flowerbed[i] == 0 && flowerbed[i-1] == 0){
                if((i+1 <= flowerbed.length-1 && flowerbed[i+1] == 0) || i+1 > flowerbed.length-1){
                    flowerbed[i] = 1;
                    count++;
                }
            }
        }
        //count与n比较，count小于等于n的时候就满足条件
        return n <= count ? true : false;
    }
}