种花问题
题目描述:
假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给你一个整数数组 flowerbed 表示花坛,由若干 0 和 1 组成,其中 0 表示没种植花,1 表示种植了花。另有一个数 n ,能否在不打破种植规则的情况下种入 n 朵花?能则返回 true ,不能则返回 false。
输入输出样例
输入:flowerbed = [1,0,0,0,1], n = 1
输出:true
题解:
本题为简单题,使用能种就种的贪心算法循环判断即可;
具体代码
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
class L605 {
public:
bool canPlaceFlowers(vector<int>& flowerbed, int n) {
int ans = 0;
for (int i = 0; i < flowerbed.size(); i++) {
//贪心算法,能种就种
if (flowerbed[i] == 0
&& (i + 1 == flowerbed.size()|| flowerbed[i + 1] == 0)
&& (i == 0 || flowerbed[i - 1] == 0)) {
flowerbed[i] = 1;
ans++;
}
}
return ans >= n;
}
};