开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第29天,点击查看活动详情
一、题目描述:
605. 种花问题 - 力扣(LeetCode) (leetcode-cn.com)
假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给你一个整数数组 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 * 10^4
- flowerbed[i] 为 0 或 1
- flowerbed 中不存在相邻的两朵花
- 0 <= n <= flowerbed.length
二、思路分析:
- 第一次循环直接从[1, len(flowerbed)-1]会有边界无法判断的情况,例如【001010】就会漏掉第一个0,所以想到给他的头和尾都加上一个0,就不用修改判断条件了:所以,当判断条件有不同的情况时,可以想想能不能不把情况区分开,而是考虑不同的情况转化成可以用同一标准来处理
- 实现逻辑:如果一个位置的前后都为零,并且他也为零,则计数加一改变当前状态,否则下一层循环
三、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
范文参考: