假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给你一个整数数组 flowerbed
表示花坛,由若干 0
和 1
组成,其中 0
表示没种植花,1
表示种植了花。另有一个数 n
,能否在不打破种植规则的情况下种入 n
朵花?能则返回 true
,不能则返回 false
。
贪心算法
判断能否在不打破种植规则的情况下在花坛内种入 n 朵花,从贪心的角度考虑,应该在不打破种植规则的情况下种入尽可能多的花,然后判断可以种入的花的最多数量是否大于或等于 n。
bool canPlaceFlowers(int* flowerbed, int flowerbedSize, int n) {
int count=0;
int prev=-1;
for(int i=0;i<flowerbedSize;i++)
{
if(flowerbed[i]==1)
{
if(prev<0)
count+=i/2;
else count+=(i-prev-2)/2;
prev=i;
}
}
if(prev<0)
count+=(flowerbedSize+1)/2;
else
count+=(flowerbedSize-prev-1)/2;
return count>=n;
}
下面提供python的写法
class Solution:
def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:
flowerbed = [0] + flowerbed + [0]
for i in range(1, len(flowerbed) - 1):
if flowerbed[i - 1] == 0 and flowerbed[i] == 0 and flowerbed[i + 1] == 0:
flowerbed[i] = 1 # 种花!
n -= 1
return n <= 0
具体可参考leetcode官网详解
链接:https://leetcode.cn/problems/can-place-flowers/solutions/542556/chong-hua-wen-ti-by-leetcode-solution-sojr/