1. 程式人生 > >leetcode 605. 種花問題(陣列)(Easy)

leetcode 605. 種花問題(陣列)(Easy)

題目:

假設你有一個很長的花壇,一部分地塊種植了花,另一部分卻沒有。可是,花卉不能種植在相鄰的地塊上,它們會爭奪水源,兩者都會死去。

給定一個花壇(表示為一個數組包含0和1,其中0表示沒種植花,1表示種植了花),和一個數 。能否在不打破種植規則的情況下種入 朵花?能則返回True,不能則返回False。

示例 1:

輸入: flowerbed = [1,0,0,0,1], n = 1
輸出: True

示例 2:

輸入: flowerbed = [1,0,0,0,1], n = 2
輸出: False

注意:

  1. 陣列內已種好的花不會違反種植規則。
  2. 輸入的陣列長度範圍為 [1, 20000]。
  3. n 是非負整數,且不會超過輸入陣列的大小。

思路:

        遍歷陣列,直到滿足可以種植一朵花的條件(陣列當前位置=0,且前後相鄰數也是0)。遍歷完整個陣列後,若❀的數目小於等於零,那麼證明❀已經全部種植完。

程式碼:

class Solution(object):
    def canPlaceFlowers(self, flowerbed, n):
        """
        :type flowerbed: List[int]
        :type n: int
        :rtype: bool
        """
        for i in range(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
                n-=1
        return n<=0