题目要求:
假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给你一个整数数组 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
解题思路
第一种思路:给列表开头和末尾都添加0,然后使用和中间3个0判断的方法一样,通过将计数n递减来判断其值是否为<=0返回true或者false。需要注意的是,需要对当前需要种花的位置填充1
代码:
func canPlaceFlowers(flowerbed []int, n int) bool {
// 开头和末尾添加0,之后可以使用是否连续3个零来判断个数
flowerbed = append(append([]int{0}, flowerbed...), 0)
for i := 1; i < len(flowerbed)-1; i++ {
if flowerbed[i-1] == 0 && flowerbed[i] == 0 && flowerbed[i+1] == 0 {
// 这步是为了填补种花的空缺,不然对于4个0的情况有问题
flowerbed[i] = 1
n--
}
if n <= 0 {
return true
}
}
return n <= 0
}
第二种思路:第二种的变种,不需要补0,直接把收尾2个0的情况考虑在内。需要注意的是,需要对当前需要种花的位置填充1,这个效率比第二种高一点。
代码:
func canPlaceFlowers(flowerbed []int, n int) bool {
for i := 0; i < len(flowerbed); i++ {
if (i == 0 || flowerbed[i-1] == 0) && flowerbed[i] == 0 && (i == len(flowerbed)-1 || flowerbed[i+1] == 0) {
flowerbed[i] = 1
n--
if n <= 0 {
return true
}
}
}
return n <= 0
}
第三种思路:这道题的难点在于如何判断在不打破种植规则的情况下种入 n 最开始我以为只要满足中间连续0的个数>=2*n+1即可,却忘考虑[1,0,0,0,1,0,0]和[0,0,1,0,1]这种0开头和0结尾的,提交了2次均是错的。看了官方题解发现用的贪心算法。计算出数组中最多能种植多少花,记为count,用count和n作比较,如果count>=n,则表示能种植,返回true,反之返回false
代码:
//还没完全理解,等后续补充