贪心-LeetCode605. 种花问题

1、题目描述

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

给定一个花坛（表示为一个数组包含0和1，其中0表示没种植花，1表示种植了花），和一个数 n 。能否在不打破种植规则的情况下种入 n 朵花？能则返回True，不能则返回False。

1. 数组内已种好的花不会违反种植规则。
2. 输入的数组长度范围为 [1, 20000]。
3. n 是非负整数，且不会超过输入数组的大小。

2、代码详解（贪心）

我们从左到右扫描数组 flowerbed，如果数组中有一个 0，并且这个 0 的左右两侧都是 0，那么我们就可以在这个位置种花，即将这个位置的 0 修改成 1，并将计数器 count 增加 1。

对于数组的第一个和最后一个位置，我们只需要考虑一侧是否为 0。

在扫描结束之后，我们将 count 与 n 进行比较。如果 count >= n，那么返回 True，否则返回 False。

class Solution(object):
    def canPlaceFlowers(self, flowerbed, n):
        """
        :type flowerbed: List[int]
        :type n: int
        :rtype: bool
        """
        # flowerbed = [0] + flowerbed +[0]
        # i = 1
        # count = 0
        # while i <= len(flowerbed)-1:
        #     if flowerbed[i-1:i+2] == [0, 0, 0]:
        #         count += 1
        #         i += 2
        #     else:
        #         i += 1
        # return n <= count
        i = 0
        count = 0
        while i < len(flowerbed):
            if flowerbed[i] == 0 and (i == 0 or flowerbed[i-1] == 0) and (i == len(flowerbed)-1 or flowerbed[i+1] == 0):
                flowerbed[i] = 1
                count += 1
            i += 1
        return count >= n

类似题，求出还可以种几棵花，返回的是原题的count

输入g是地形、n是g的长度

def tree(g, n):
    '''
    :param g: list
    :param n: int
    :return:
    '''
    if n != len(g):
        return -1
    i = 0
    count = 0
    while i < len(g):
        if g[i] == 0 and (i == 0 or g[i - 1] == 0) and (
                i == len(g) - 1 or g[i + 1] == 0):
            g[i] = 1
            count += 1
        i += 1
    return count
n=int(input())
g = list(map(int, input().split()))
ans = tree(g, n)
print(g)
print(ans)