题目:
假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给你一个整数数组 flowerbed 表示花坛,由若干 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 <= flowerbed.length <= 2 * 104
- flowerbed[i] 为 0 或 1
- flowerbed 中不存在相邻的两朵花
- 0 <= n <= flowerbed.length
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/can-place-flowers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
就是当前位置的左右相邻位置都没有种花(为0),那么当前位置就为种花(设为1),能种花存在以下几种情况:
- 当前位置flowerbed[i] == 0,并且当前位置是数组的末尾:i = flowerbed.length - 1,并且它的前一个位置没有种花:flowerbed[i -1] == 0;
- 当前位置flowerbed[i] == 0,并且当前位置是数组第一个:i == 0,并且它的下一个位置没有种花:flowerbed[i + 1] == 0;
- 当前位置flowerbed[i] == 0,并且数组只有一个元素,i == 0并且 i = flowerbed.length - 1;
- 当前位置flowerbed[i] == 0,并且它的前一位置没有种花,flowerbed[i -1] == 0,并且它的后一位置也没有种花:flowerbed[i +1] == 0;
java代码:
1 class Solution { 2 public boolean canPlaceFlowers(int[] flowerbed, int n) { 3 int m = flowerbed.length; 4 int count = 0; 5 for(int i = 0; i < m; i++){ 6 if(flowerbed[i] == 0 && (i == m-1 || flowerbed[i+1] == 0) && (i == 0 || flowerbed[i-1] == 0)){ 7 flowerbed[i] = 1; 8 count ++; 9 } 10 } 11 return count >= n; 12 } 13 }
python3代码:
1 class Solution: 2 def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool: 3 m = len(flowerbed) 4 count = 0 5 for i in range(m): 6 if flowerbed[i] == 0 and (i == 0 or flowerbed[i - 1] == 0) and (i == m - 1 or flowerbed[i + 1] == 0): 7 flowerbed[i] = 1 8 n -= 1 9 return n <= 0标签:605,count,位置,java,python,flowerbed,int,种花,当前 From: https://www.cnblogs.com/liu-myu/p/16811782.html