605. 种花问题
时间:2021年1月1日
知识点:贪心
题目链接
题目
假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给你一个整数数组 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
解法:
- 贪心算法 自己模拟一遍
- 如果当前位置种花了 至少在i+2的位置才能种花
- 如果当前位置没种花 且下个位置也没种花 这个位置可以种花 注意边界的情况
- 否则 那么至少在i+3的位置才能种花
代码
//头文件
#include "cheader.h"
class Solution {
public:
bool canPlaceFlowers(vector<int>& flowerbed, int n) {
int all = flowerbed.size();
for(int i = 0;i < all;){
if(flowerbed[i] == 1)
i+=2;
else if(flowerbed[i] == 0){
if(i == all - 1 || flowerbed[i+1] ==0){
n--;
i+=2;
}
else
i+=3;
}
}
return n <= 0;
}
};
int main()
{
vector<int> flowerbed{1,0,0,0,1};
Solution s;
int n = 1;
cout<<s.canPlaceFlowers(flowerbed, n)<<endl;
return 0;
}
今天也是爱zz的一天哦!