5438. 制作 m 束花所需的最少天数
给你一个整数数组 bloomDay,以及两个整数 m 和 k 。
现需要制作 m 束花。制作花束时,需要使用花园中 相邻的 k 朵花 。
花园中有 n 朵花,第 i 朵花会在 bloomDay[i] 时盛开,恰好 可以用于 一束 花中。
请你返回从花园中摘 m 束花需要等待的最少的天数。如果不能摘到 m 束花则返回 -1 。
思路:
判断是否能制作成指定数量的花束,这里直接遍历,计算连续花朵数量,达到k朵,就将花束加一。
int minDays(vector<int>& bloomDay, int m, int k) {
int len = bloomDay.size();
if(m*k > len)
{
return -1;
}
int r = len-1;
int l = 0;
vector<int> rec(bloomDay);
sort(rec.begin(), rec.end()); // 排序,使用二分法
while(l<= r)
{
int mid = (l+r)/2;
if(check(bloomDay, m, k, rec[mid]))
{
r = mid-1;
}
else
{
l = mid+1;
}
}
return rec[l];
}
bool check(vector<int>& bloomDay, int m, int k, int day)
{
int len = bloomDay.size();
//贪心法
int count = 0; //连续花朵数目
int cur = 0; //花束数量
for(int i = 0; i < len; ++i)
{
if(bloomDay[i] <=day)
{
//开花了
count++;
if(count == k)
{
//满足制成花束
cur++;
count = 0;
}
if(cur == m)
{
return true;
}
}
else
{
count = 0;
}
}
return false;
}