日新计划605. 种花问题

81 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第28天,点击查看活动详情

一、题目描述:

605. 种花问题 - 力扣(LeetCode)

假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。

给你一个整数数组  flowerbed 表示花坛,由若干 01 组成,其中 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 * 10^4
  • flowerbed[i] 为 0 或 1
  • flowerbed 中不存在相邻的两朵花
  • 0 <= n <= flowerbed.length

二、思路分析:

问题为一个一维数组,种花要求前后都不是1,第一个数和最后一个数较为特别,只需附近的一个位置不为1即可;
所以可分为三类,第一类:普通情况 ————所有都要求当前数值不为1; 第二类 首位置————位置索引为0,否则前一个位置不为1; 第三类 末位置———— 位置索引为总长度减一,否则后一个位置不为一;利用for循环,判断三类情况的并集;成功情况下,将当前位置赋值为1,避免影响后续判断;同时将位置索引额外加一,可减少查询次数,节省时间;最后将预计种的花数n减一;最后若n小于等于零,则当前数组可钟预计花数。

三、AC 代码:

bool canPlaceFlowers(int* flowerbed, int flowerbedSize, int n){
for(int i=0;i<flowerbedSize;i++){//利用变量i遍历flowerbed数组
    if(flowerbed[i]!= 1 &&(i==0 || flowerbed[i-1]!=1)&&(i==flowerbedSize-1 || flowerbed[i+1]!=1))//如果当前数值不为1,且为第一个位置的数值或当前位置的前一个数值不为1,同时为最后一个位置的数值或当前位置的下一个位置数值不为1,则当前位置可种花
    {   
        flowerbed[i] = 1;//为当前位置种花赋值
        i++;//当前位置种花,则下个位置必定不能种花,额外加一
        n--;//令要求种花数减一
    }
}
return n <= 0;//若当前n的值小于等于一,即可钟n多花,返回true,反之则返回false
}