leetcode 605. 種花問題(陣列)(Easy)
阿新 • • 發佈:2018-11-04
題目:
假設你有一個很長的花壇,一部分地塊種植了花,另一部分卻沒有。可是,花卉不能種植在相鄰的地塊上,它們會爭奪水源,兩者都會死去。
給定一個花壇(表示為一個數組包含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, 20000]。
- 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