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

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;
}
}