Leetcode 605. 种花问题(终于解决)
假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给你一个整数数组 flowerbed 表示花坛,由若干 0 和 1 组成,其中 0 表示没种植花,1 表示种植了花。另有一个数 n ,能否在不打破种植规则的情况下种入 n 朵花?能则返回 true ,不能则返回 false。
示例 1:
输入:flowerbed = [1,0,0,0,1], n = 1
输出:true
示例 2:
输入:flowerbed = [1,0,0,0,1], n = 2
输出:false
提示:
- 1 <= flowerbed.length <= 2 * 104
- flowerbed[i] 为 0 或 1
- flowerbed 中不存在相邻的两朵花
- 0 <= n <= flowerbed.length
主要思路:如果出现两个元素相同且均为零,然后分别判断该元素左边和右边是为均为1,这里有个易错点,一定要先判断左边元素是否满足种花条件
Code:
class Solution {
public:
bool canPlaceFlowers(vector<int>& flowerbed, int n) {
int cnt=0;
if(flowerbed.size()==1&&flowerbed[0]==0)
{
if(n==1)
return true;
}
for(int i=1;i<flowerbed.size();i++)
{
if(flowerbed[i]==flowerbed[i-1]&&(flowerbed[i]==0))
{
//先判断左边是否满足条件
if((i-2)<0)
{
flowerbed[i-1]=1;
cnt++;
continue;
}
if((i-2)>=0&&flowerbed[i-2]==0)
{
flowerbed[i-1]=1;
cnt++;
continue;
}
//再判断左边是否满足条件
if((i+1)<flowerbed.size()&&(i-1)>=0)
{
if(flowerbed[i+1]==0&&flowerbed[i-1]==0)
{
flowerbed[i]=1;
cnt++;
continue;
}
}
if((i+1)>=flowerbed.size())
{
flowerbed[i]=1;
cnt++;
continue;
}
}
}
if(cnt>=n)
return true;
return false;
}
};