题目描述
给你一个整数数组 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
示例 2:
输入:bloomDay = [1,10,3,10,2], m = 3, k = 2
输出:-1
解释:要制作 3 束花,每束需要 2 朵花,也就是一共需要 6 朵花。而花园中只有 5 朵花,无法满足制作要求,返回 -1 。
示例 3:
输入:bloomDay = [7,7,7,7,12,7,7], m = 2, k = 3
输出:12
解释:要制作 2 束花,每束需要 3 朵。
花园在 7 天后和 12 天后的情况如下:
7 天后:[x, x, x, x, _, x, x]
可以用前 3 朵盛开的花制作第一束花。但不能使用后 3 朵盛开的花,因为它们不相邻。
12 天后:[x, x, x, x, x, x, x]
显然,我们可以用不同的方式制作两束花。
示例 4:
输入:bloomDay = [1000000000,1000000000], m = 1, k = 1
输出:1000000000
解释:需要等 1000000000 天才能采到花来制作花束
示例 5:
输入:bloomDay = [1,10,2,9,3,8,4,7,5,6], m = 4, k = 2
输出:9
提示:
bloomDay.length == n
1 <= n <= 10^5
1 <= bloomDay[i] <= 10^9
1 <= m <= 10^6
1 <= k <= n
解题思路
首先从题目中分析,那些场景可以满足要求,那些场景不可以满足?
实际上,只要有m*k朵花,就一定有解。因此,从给定数据的个数上,就可以判断出此组数据是否有解。
接下来考虑这个题只剩下一个疑问需要解答,那就是怎么找到最少的天数?
先考虑两个极端情况:
1.最少天数是什么场景?
2.最多天数是什么场景?
接下来根据这两个问题进行解析:
1.最少天数,如果是需要0朵花,那岂不是0天就搞定了?观察输入数据发现,不可能是0朵花。所以再考虑如果1朵花呢?假设只有一朵花,那么最短时间就是开花时间最短的那一朵花所需要的时间。
2.最多的天数,最慢情况下,那就是所有花全都开了,这样花费的时间实际上就是开花最晚的那一朵所需要的时间。
结合上面的分析,可以得出结论:题目的答案一定在开花最快和最慢两朵花所需时间范围内。既然题目答案的范围已经确定,那直接遍历这个时间,就一定可以找到题目的解。但实际上这是一个比较耗时的方法,这么做会导致超时。需要更快速度的方法,既然范围都知道了,那也就可以用二分查找来找到合适的时间了。每次找到一个时间,检查一下开的花能否满足题目要求。下面看C++代码
代码
int minDays(vector<int>& bloomDay, int m, int k)
{
int len = bloomDay.size();
int minDay = *min_element(bloomDay.begin(), bloomDay.end()); // 开得最快的一朵花需要的时间
int maxDay = *max_element(bloomDay.begin(), bloomDay.end()); // 开得最慢的一朵花需要的时间
// 花不够。总数都不够,怎么凑都不行,直接 -1
if (len < m * k)
{
return -1;
}
// 数量刚刚好,那直接用开得最慢的那一朵花需要的时间,因为要全部都开,才可以凑够要求
else if(len == m * k)
{
return maxDay;
}
// 其他情况:只要花的数量达标,那么,就算花开得再晚,一定会满足定条件的。
// 所以接下来就是一个时间长短的问题。
// 要求时间尽可能短的满足条件。最短时间是多少?想象一种极端情况,开了一朵花,就满足条件。那这个时间是多少?
// 最短时间一定是开得最快的那一朵花需要的时间!!!会不会开0朵就满足了呢?查看题目输入要求,都是大于0的,所以不会。
// 哪最长时间要多少?假设现在需要所有花都开完才能满足。。。(这个就是上面的else if分支)最长时间一定是开得最慢的那一朵花需要的时间。
// 因为等最慢的花开了,所有花都开完了,时间再继续走下去,也没有新的花要开。。。岂不是浪费时间???
// 综上所述,得出一个结论:题目的答案一定在minDay和maxDay这两个数之间!
// 既然都能确定时间范围了,这就好办了,最简单暴力的方式:我把这两个日期中间所有时间遍历一下,一定能得出结果。但速度不行,更好的办法就是把日期二分查找。
// 下面开始操作起来
// 用于检查是否满足条件,lambda
auto isOK = [&](int nowDay) -> bool {
int cnt = 0;
int total = 0;
for (int &x : bloomDay)
{
if(x > nowDay)
{
cnt = 0;
}
else
{
cnt++;
}
if(cnt == k)
{
total++;
cnt = 0;
}
}
return total >= m;
};
// 这里开始二分
int left = minDay;
int right = maxDay;
int mid = 0;
while (left <= right)
{
mid = (left + right) / 2;
if (isOK(mid))
{
//满足条件了,还不能返回,因为不一定是最短时间的,这里修改右侧边界,继续找
right = mid - 1;
}
else
{
left = mid + 1;
}
}
return right + 1; // right里面总是记录着最后一次满足条件的位置,因为上面减了1,所以这里加回来
}