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]
is0
or1
.- There are no two adjacent flowers in
flowerbed
. 0 <= n <= flowerbed.length
My answer:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | class Solution { public boolean canPlaceFlowers(int[] flowerbed, int n) { if (n == 0) { return true; } // greedy should be enough for (int i = 0; i < flowerbed.length; i ++) { if (canPlace(i, flowerbed)) { flowerbed[i] = 1; n--; if (n == 0) { return true; } } } return false; } private boolean canPlace(int i, int[] flowerbed) { return flowerbed[i] == 0 && flowerbed[Math.max(0, i - 1)] == 0 && flowerbed[Math.min(flowerbed.length - 1, i + 1)] == 0; } } |
No comments:
Post a Comment