题目:
假设你有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花卉不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给定一个花坛(表示为一个数组包含0和1,其中0表示没种植花,1表示种植了花),和一个数 n 。能否在不打破种植规则的情况下种入 n 朵花?能则返回True,不能则返回False。
思路:
根据题目,可以将数组分成三个段,首,中部,尾,首只要第一项和第二项是0,就可以种花,尾需要最后一项和倒数第二项为0就可以种花,中部需要连续3个是0,中间可以种花。
sum用来记录能够种花的地方,最后与目标n进行比较,只要<=n就表示可以。
代码:
class Solution {
public boolean canPlaceFlowers(int[] flowerbed, int n) {
int sum=0;
if(flowerbed.length==1)
{
if(flowerbed[0]==0&&n==1)
return true;
}
else
{
if(flowerbed[0]==0&&flowerbed[1]==0)
{ flowerbed[0]=1;
sum++;}
for(int i=1;i<flowerbed.length-3;i++)
{
if(flowerbed[i]==0&&flowerbed[i+1]==0&&flowerbed[i+2]==0){
flowerbed[i+1]=1;
sum++;
}
}
int k=flowerbed.length;
if(flowerbed[k-2]==0&&flowerbed[k-1]==0)
{
flowerbed[k-1]=1;
sum++;
}
}
if(sum>=n)
return true;
else
return false;
}
}