LeetCode: 605-Can Place Flowers 解題紀錄

Last Updated on 2022-01-18 by Clay

題目

You have a long flowerbed in which some of the plots are planted, and some are not. However, flowers cannot be planted in adjacent plots.

Given an integer array flowerbed containing 0's and 1's, where 0 means empty and 1 means not empty, and an integer n, return if n new flowers can be planted in the flowerbed without violating the no-adjacent-flowers rule.

Example 1:

Input: flowerbed = [1,0,0,0,1], n = 1
Output: true

Example 2:

Input: flowerbed = [1,0,0,0,1], n = 2
Output: false

Constraints:

• 1 <= flowerbed.length <= 2 * 104
• flowerbed[i] is 0 or 1.
• There are no two adjacent flowers in flowerbed.
• 0 <= n <= flowerbed.length

題目給定一個花壇（flowerbed）的列表，列表中只會包含 0 或 1。0 的情況代表著這裡沒有種花、1 的情況則代表這裡有種花。而另外還會有一個輸入值 n 代表著我們還要種多少朵花。

花朵不能相鄰地去種（大概是考慮到土壤養分）。

我們要判斷的，就是我們是否能以現有的花壇空間，種植下 n 朵花。可以的情況回傳 true、不行的情況回傳 false。

---

解題思路

我的解法非常地單純，就是從花壇的第一筆空間開始計算起。

• 如果花壇的 i-1 空間存在（亦即當前 i 不是第一筆資料），則判斷是否有空位，無空位的情況則跳過這次的判斷，因為無法種植
• 如果花壇的 i+1 空間存在（亦即當前 i 不是最後一筆資料），則判斷是否有空位，無空位的情況則跳過這次的判斷，因為無法種植
• 除此之外，只要當前 i 的空間為 0，則可以種植，將所需種植的 n 朵目標減去 1 並把 flowerbed[i] 設為 1；當 n 小於等於 0 的情況，則返回 true
• 如果到遍歷整個花壇結束 n 仍然大於 0，則返回 false

複雜度

Time Complexity	O(n)
Space Complexity	O(1)

C++ 範例程式碼

class Solution {
public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        for (int i=0; i<flowerbed.size(); ++i) {
            if (i-1 >= 0 && flowerbed[i-1] == 1) continue;
            if (i+1 < flowerbed.size() && flowerbed[i+1] == 1) continue;
            if (flowerbed[i] == 0) {
                flowerbed[i] = 1;
                --n;
            }
            
            if (n <= 0) return true;
        }
        
        return false;
    }
};

Python 範例程式碼

class Solution:
    def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:
        for i in range(len(flowerbed)):
            if i-1 >= 0 and flowerbed[i-1] == 1: continue
            if i+1 < len(flowerbed) and flowerbed[i+1] == 1: continue
            
            if (flowerbed[i] == 0):
                n -= 1
                flowerbed[i] = 1
            
            if n <= 0: return True
        
        return False

---

References

• https://leetcode.com/problems/can-place-flowers/

---

Read More

• LeetCode 解題紀錄目錄