问题描述
给你一个整数数组 bloomDay,以及两个整数 m 和 k 。
现需要制作 m 束花。制作花束时,需要使用花园中 相邻的 k 朵花 。
花园中有 n 朵花,第 i 朵花会在 bloomDay[i] 时盛开,恰好 可以用于 一束 花中。
请你返回从花园中摘 m 束花需要等待的最少的天数。如果不能摘到 m 束花则返回 -1 。
示例 1:
输入:bloomDay = [1,10,3,10,2], m = 3, k = 1
输出:3
解释:让我们一起观察这三天的花开过程,x 表示花开,而 _ 表示花还未开。
现在需要制作 3 束花,每束只需要 1 朵。
1 天后:[x, _, _, _, _] // 只能制作 1 束花
2 天后:[x, _, _, _, x] // 只能制作 2 束花
3 天后:[x, _, x, _, x] // 可以制作 3 束花,答案为 3
解题报告
很明显,用二分法。
实现代码
class Solution {
public:
int minDays(vector<int>& bloomDay, int m, int k) {
if(m*k>bloomDay.size()) return -1;
int n=bloomDay.size();
int l=0,r=*max_element(bloomDay.begin(), bloomDay.end()), mid;
while(l<=r){
mid=l+(r-l)/2;
int result =judge(mid, m, k, bloomDay);
if(result) r=mid-1;
else l=mid+1;
}
return l;
}
int judge(int target, int m, int k, vector<int>& bloomDay){
int counter=0, window=0;
for(int i=0;i<bloomDay.size();i++){
if(bloomDay[i]<=target) window++;
else window=0;
if(window==k){
window=0;
counter++;
}
}
return counter>=m;
}
};