题目大意:
给你一个整数数组 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
解题思路:
如果需要的总束数量乘以每束花所需的朵数,超过了花园总数,则无法制作,返回-1;
对天数二分搜索,如果如果第mid天能够制作,则继续寻找更小的天数:
class Solution {
public:
int minDays(vector<int>& bloomDay, int m, int k) {
int total = bloomDay.size();
if (m * k > total){
return -1;
}
int left = 0, right = 1e9, mid;
while (left < right){
mid = left + (right - left) / 2;
// 如果返回了true,则表示用mid天一定能制成m束花,所以再去找更小的天数
if (canGet(mid, bloomDay, m, k)){
right = mid;
}
else{
left = mid + 1;
}
}
return left;
}
bool canGet(int mid, vector<int> & bloomDay, int m ,int k){
// 能够制作的花束,当前能做成tmpday长度的花
int yesDay = 0, tmpDay = 0;
// 从头遍历,制作花束
for (int i = 0; i < bloomDay.size(); ++i){
// 如果当前的花束成长时间少于指定天数,则表示位置i的花可以用来制作
if (bloomDay[i] <= mid){
++tmpDay;
}
// 反之,则不能用i位置来制作,因此重新计数
else{
tmpDay = 0;
}
// 如果计数满足了一束花的所需,则制成花束+1, 后面从0开始重新计数
if (tmpDay == k){
yesDay++;
tmpDay = 0;
}
// 如果制成了m朵花,则返回true
if (yesDay == m){
return true;
}
}
return false;
}
};