题目:
假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给你一个整数数组 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
分析
花不能种植在相邻的地块上,也就意味着不能有连续的2个1,我们判断每个坑位能不能种无非就是看它自身的元素以及前面的元素和后面的元素是不是都是0,如果连续3个都是0那就代表可以种,用这种思路有个问题就是如果已经种过的会影响下一个位置能不能种,以及有些边界条件也需要处理。
我们可以把问题简化一下,不依次遍历每个元素,只看当前元素和下一个元素是否都是0,如果当前元素和下一个元素都是0,那么当前元素肯定可以是1,然后当前元素的下一个元素就可以不用遍历了,因为它肯定不能是1
public class canPlaceFlowers {
public static void main(String[] args) {
int[] arr = {1,0,0,0,1};
System.out.println(canPlaceFlower(arr,1));
}
public static boolean canPlaceFlower(int[] arr,int n) {
int len = arr.length;
for(int i = 0;i<len && n>0;) {
if(arr[i] == 1) {
i+=2;
} else if(i+1==len || arr[i+1] == 0) {
n--;
i+=2;
} else {
i+=3;
}
}
return n <=0;
}
}