https://leetcode-cn.com/problems/can-place-flowers/
假设你有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花卉不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给定一个花坛(表示为一个数组包含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, 20000]。
n 是非负整数,且不会超过输入数组的大小。
/* 思路
遍历数组,查找连续的0(>=三个)首尾(>=2)个的0
计算可插入个数 不算首尾连续的n个0可以插入的个数为(n - 1)/2 , 首尾的为 n / 2
比较复杂
思路二:前后加上 10....01,统一计算,不算头尾的特殊情况
*/
void init_new_arr(int* flowerbed, int flowerbedSize, int *new_arr, int len){
new_arr[0] = 1;
new_arr[1] = 0;
int i = 0;
for(; i < flowerbedSize; i++){
new_arr[i + 2] = flowerbed[i];
}
new_arr[len - 2] = 0;
new_arr[len - 1] = 1;
}
bool canPlaceFlowers(int* flowerbed, int flowerbedSize, int n){
int len = flowerbedSize + 4;
int *new_arr = (int *)malloc(len * sizeof(int));
init_new_arr(flowerbed, flowerbedSize, new_arr, len);
int i = 0;
int j = 0;
int count = 0;
int sum = 1; // 此处考虑第一个为1的情况
for(; i < len; i++){
if(1 == new_arr[i]){
count = i - j - 1;
sum += (count - 1) / 2;
count = 0;
j = i;
}
if(0 == new_arr[i]){
count++;
}
}
return n <= sum ? true : false;
}