605_can_place_flowers

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

代码参考：

package main

import "fmt"

func main() {
    fmt.Println(canPlaceFlowers([]int{0}, 1)) // true
}

// 种花的本质是标识 0，即是一边遍历一边修改为 1
func canPlaceFlowers(flowerbed []int, n int) bool {
    length := len(flowerbed)
    if length == 1 && flowerbed[0] == 0 {
        return true
    }

    // first blood
    if flowerbed[0] == 0 && flowerbed[1] == 0 {
        flowerbed[0] = 1 // 种花
        n--
    }
    for i := 1; i < length; i++ {
        // 前后都是 0 就可以种花，注意最后一位的 case
        if flowerbed[i-1] == 0 && flowerbed[i] == 0 && (i == length-1 || flowerbed[i+1] == 0) {
            flowerbed[i] = 1
            n--
        }
    }
    return n <= 0
}