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;
        
        
    }
};
posted @ 2022-05-31 11:00  萧海~  阅读(24)  评论(0编辑  收藏  举报