题目
假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给你一个整数数组 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时,根据题意,下一位一定为0,因此可以直接i=i+2;
当当前为0时,前一位必定为0或者该位为首位。因为如果前一位为1的话,上次的循环会直接+2,不会出现前位是1的情况,因此只需要判断下一位是不是0,若是0或者为最后一位,因为前位和当前为都是0,因此可以种花,种完花以后该位变为1,因此只需要跳两位。当下一位为1时,可以直接让i跳3位。
class Solution {
public:
bool canPlaceFlowers(vector& flowerbed, int n) {
int num = flowerbed.size();
for(int i=0;i<num;i+=2){
if(flowerbed[i]0){
if(inum-1 || flowerbed[i+1]==0){
n–;
if(n<=0) break; //优化循环次数
}
else{ //下一位为1
i++;
}
}
}
return n<=0;
}
};