leetcode605 种花问题

最新推荐文章于 2025-01-29 23:39:44 发布

原创最新推荐文章于 2025-01-29 23:39:44 发布 · 149 阅读

0 ·

CC 4.0 BY-SA版权

博客围绕花坛种花问题展开，给定由0和1组成的整数数组表示花坛，0表示未种花，1表示已种花，还有一个数n。需判断能否在不打破种植规则下种入n朵花。采用贪心算法，通过遍历数组计算可种花数量，当超过n时输出结果，还考虑了花坛无花的特殊情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

假设有一个很长的花坛，一部分地块种植了花，另一部分却没有。可是，花不能种植在相邻的地块上，它们会争夺水源，两者都会死去。
给你一个整数数组 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

思路：
感觉像是贪心算法。。。先设定一个记录最近的花位置的变量flower，然后从头至尾遍历，设遍历到花的下标为i，第一个遇到花时可以在前面种i/2朵花，然后将flower更新为i，第二个遇到花时可以在中间种(flower-i)/2朵花，再次更新i。。。最后结束遍历，根据此时的flower和总长度size，还可以种（size-1-flower）/2朵花。然而这里有个特殊情况，花坛里没有种花，则一共可以种（size+1）/2朵花。由于我们需要给flower一个初值从而确定第一次遇到花的位置，我们可以利用这个初值使得最后的处理不用分类讨论。即令flower=-2。另外，由于本题并不要求求出最多可以种植的花，当花的数量超过n，即可输出结果。

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
int size=flowerbed.size();
int count=0;
int flower=-2;
for(int i=0;i<size;i++)
{
    if(flowerbed[i]==1)
    {
        if(flower<0)
        {
            count=count+i/2;
        }
        else
        {
            count=count+(i-flower-2)/2;
        }
        if(count>=n)
        {
            return true;
        }
        flower=i;
    }
}
count=count+(size-1-flower)/2;
    if(count>=n)
        {
            return true;
        }
        else return false;
    }
};