Leetcode605.Can Place Flowers種花問題
阿新 • • 發佈:2018-11-06
假設你有一個很長的花壇,一部分地塊種植了花,另一部分卻沒有。可是,花卉不能種植在相鄰的地塊上,它們會爭奪水源,兩者都會死去。
給定一個花壇(表示為一個數組包含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 是非負整數,且不會超過輸入陣列的大小。
class Solution { public: bool canPlaceFlowers(vector<int>& flowerbed, int n) { int len = flowerbed.size(); int sum1 = 0; int sum2 = 0; for(int i = 0; i < len; i = i + 2) { if(i == 0 && flowerbed[i + 1] != 1 && flowerbed[i] != 1) sum1++; else if(i == len - 1 && flowerbed[i - 1] != 1 && flowerbed[i] != 1) sum1++; else if(flowerbed[i + 1] != 1 && flowerbed[i - 1] != 1 && flowerbed[i] != 1) sum1++; } for(int i = 1; i < len; i = i + 2) { if(flowerbed[i + 1] != 1 && flowerbed[i - 1] != 1 && flowerbed[i] != 1) sum2++; } if(sum1 >= n || sum2 >= n) return true; return false; } };