貪心-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)