种花问题(力扣)
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
解题思路(贪心算法)
贪心思路
在这种情况下,我们可以遍历花坛数组 flowerbed,并在每个未种植花的
地块上尝试种植花,同时确保不违反相邻地块不得种植的规则。
1、遍历整个 flowerbed 数组。
2、对于每个元素,检查它是否为 0(即没有种植花), 并且它的相邻地块(如果存在)也没有种植花。
3、如果可以种植花,将该位置设为 1,并将 n 减 1。
4、如果 n 减到 0 或以下,返回 true。
5、遍历结束后,如果 n 仍大于 0,则返回 false。
在代码实现中,循环是从索引 1 开始,直到 N-1 结束。这意
味着它不考虑花坛数组的第一个和最后一个位置。在一些情
况下,如果花坛的两端没有种植花,它们也是可以种植新花的合适位置。
所以在循环之前和之后检查花坛的两端。
C++代码片
下面展示为C++风格的 代码片
.
class Solution {
public:
bool canPlaceFlowers(vector<int>& flowerbed, int n) {
int N=flowerbed.size();
// 检查开始位置
if (N > 0 && flowerbed[0] == 0 && (N == 1 || flowerbed[1] == 0)) {
flowerbed[0] = 1;
n--;
}
// 检查最后位置
if (flowerbed[N-1] == 0 && flowerbed[N-2] == 0) {
flowerbed[N-1] = 1;
n--;
}
for(int i=1;i<N-1;i++)
{
if(flowerbed[i]==0&&flowerbed[i-1]==0&&flowerbed[i+1]==0)
{
n--;
flowerbed[i]=1;
if (n <= 0)
return true;
}
}
return n<=0;
}
};
首先检查花坛的第一个位置是否可以种植花,然后遍历中间的位置,最后
检查最后一个位置。注意,如果花坛数组只有一个位置,那么这个位置既
是第一个也是最后一个位置,因此只需检查一次。