分为多种情况:
1)两边都有花
假设两个花的位置分别为x,y,中间连续的0的个数为(x-y-1)个然后要去掉与这两朵花向邻的0,所以中间剩下的0的个数为(x-y-3),至少要剩下一个0才可以种花,所以在中间能够种花的条件为x-y>=4,
那么中间能够种几朵花呢?
如果有奇数个0,那么能够种花的数目为(x-y-2)/2
如果有偶数个0,那么能够种花的数目为(x-y-3)/2
但是由于整数除法向下取整的特性,这个值与(x-y-2)/2相同。
至此偶数和奇数个0求花朵的个数得到了统一。
2)如果两边没有花
那么中间能够种花的位置为0的个数比上面多2,其他的做法一样
3)如果有一端有花,一段没有花,那么可以种花的0的个数比第一种多1,其他的都一样。
如何区分这3种情况,设置一个pre,并将其初始化为负数,如果如果第一次遇到种花的地方,并且检测pre为负数,说明只有一端有花,采用第二种方法
如果为正数,说明两端有花,采用第一种方法,
并在每次操作完成后都将最新找到的花的位置赋值给pre
最后就是要考虑循环完成之后,最后一个位置是否有花
如果没有花,那么pre依然是负数,连续为0的长度是整个花坛的长度,直接用第二种方法
如果是正数,那就用第三种方法
因为-1/2是向下取整,结果为0,所以即使是pre指向花坛的最后一个位置,并且最后一个位置上面有花也不是什么大问题。
class Solution {
public:
bool canPlaceFlowers(vector<int>& flowerbed, int n) {
int count = 0;
int m = flowerbed.size();
int prev = -1;
for (int i = 0; i < m; i++) {
if (flowerbed[i] == 1) {
if (prev < 0) {
count += i / 2;
} else {
count += (i - prev - 2) / 2;
}
if (count >= n) {
return true;
}
prev = i;
}
}
if (prev < 0) {
count += (m + 1) / 2;
} else {
count += (m - prev - 1) / 2;
}
return count >= n;
}
};
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/can-place-flowers/solution/chong-hua-wen-ti-by-leetcode-solution-sojr/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。