605. 种花问题

105 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第29天,点击查看活动详情

一、题目描述:

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

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

给你一个整数数组  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, len(flowerbed)-1]会有边界无法判断的情况,例如【001010】就会漏掉第一个0,所以想到给他的头和尾都加上一个0,就不用修改判断条件了:所以,当判断条件有不同的情况时,可以想想能不能不把情况区分开,而是考虑不同的情况转化成可以用同一标准来处理
  2. 实现逻辑:如果一个位置的前后都为零,并且他也为零,则计数加一改变当前状态,否则下一层循环

三、AC 代码:

class Solution:
    def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:
        count = 0
        flowerbed = [0]+flowerbed+[0]
        for i in range(1, len(flowerbed)-1):
            if flowerbed[i-1] == 0 and flowerbed[i+1]==0 and flowerbed[i]==0:
                flowerbed[i] = 1
                count += 1
        return count>=n

范文参考

种花问题(贪心:当前位置能种花,就种但要满足几种情况☀) - 种花问题 - 力扣(LeetCode)

先将flowerbed进行扩张,使得两端不需要进行特殊处理;然后只需要考虑连续n个0可以种多少花即可--代码中的:(counterZero-1)// 2,这里需要自己拿笔推一下,然后去掉counterZero==0的情况; - 种花问题 - 力扣(LeetCode)