算法分析——蛮力法

Problem P14. [算法课蛮力法]种花问题

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

给你一个整数数组  flowerbed 表示花坛，由若干 0 和 1 组成，其中 0 表示没种植花，1 表示种植了花。另有一个数 n ，能否在不打破种植规则的情况下种入 n 朵花？能则返回 true ，不能则返回 false。

提示：

• 1 <= flowerbed.length <= 2 * 10^4
• flowerbed[i] 为 0 或 1
• flowerbed 中不存在相邻的两朵花
• 0 <= n <= flowerbed.length

输入

输入一行数组flower与需要插入的花数目

输出

输出是否插入成功

样例

标准输入复制文本

1 0 0 0 1 2

标准输出复制文本

false

class Solution:
    def canPlaceFlowers(self, flowerbed, n):
        flowerbed = [0] + flowerbed + [0]
        for i in range(1, len(flowerbed) - 1):
            if sum(flowerbed[i - 1 : i + 2]) == 0:
                flowerbed[i] = 1
                n -= 1
        return n <= 0

if __name__ == "__main__":
    solution = Solution()
    flowerbed = list(map(int, input().split()))
    n = int(input())
    result = solution.canPlaceFlowers(flowerbed, n)

    if result:
        print("true")
    else:
        print("false")