题目
假设你有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花卉不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给定一个花坛(表示为一个数组包含0和1,其中0表示没种植花,1表示种植了花),和一个数 n 。能否在不打破种植规则的情况下种入 n 朵花?能则返回True,不能则返回False。
注意:
数组内已种好的花不会违反种植规则。
输入的数组长度范围为 [1, 20000]。
n 是非负整数,且不会超过输入数组的大小。
思路
从头到尾遍历一遍,只要一个位置的左边右边都不是1且当前位置为0,那么就计数并且种花,特别要主要,当i==0的时候,可能会越界,和当i到达最后一个位置时,也会越界,做一下特殊处理。
代码
class Solution {
public boolean canPlaceFlowers(int[] flowerbed, int n) {
//种花
int length=flowerbed.length;
int count=0;
for (int i=0;i<length;++i)
{
if (flowerbed[i]==0 && (i==0 || flowerbed[i-1]==0) && (i==length-1 || flowerbed[i+1]==0))
{
flowerbed[i]=1;
++count;
}
}
return count>=n;
}
}
结果
题解的解法与这个思想类似,贪心解法
题解的优化
class Solution {
public boolean canPlaceFlowers(int[] flowerbed, int n) {
//种花
int length=flowerbed.length;
int count=0;
for (int i=0;i<length;++i)
{
if (flowerbed[i]==0 && (i==0 || flowerbed[i-1]==0) && (i==length-1 || flowerbed[i+1]==0))
{
flowerbed[i]=1;
++count;
}
//if(count>=n) 题解是在这里 利用count>=n ,其实不用,这样子,还得判断大于和等于两种情况,因为count是不断从小到大自增的,一定是先有count==n的,这样就可以返回true了
if (count==n)
return true;
}
return count>=n;
}
}