题目
1482. 制作 m 束花所需的最少天数
给你一个整数数组 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
解题思路
- 二分法。
- 通过读题可以发现答案一定是处在bloomDay的最小值和最大值之间。
- 本题的实质: 求m束花需要等待的最少的天数,所以需要在low值和high值中寻找最小的天数。
- 我们需要设定一个函数用来判断是否可以制作出m束花束。
- 我需要初始化两个值分别代码花束和花朵。
- 函数的具体实现方法:遍历数组bloomDay,如果其中的值小于等于k,那么花束的数量就加1,然后花的数量重置为0,当花束的数量等于m时,结束遍历。
- 二分法的具体操作:写出low值和high值,然后取中间的数值,如果中间的数值可以制作出m束花束(满足我们设定的函数,就代表中间的值或许可以再缩小或许就是答案),那么high值就等于中间的数值再继续计算。否则的话,low值等于中间的数值加1.
- 细节操作:每次制作出一个花束后,需要把花的数量重置为0,或者没有连着的k个花可以制作花束,也需要把花的数量重置为0.
Code
class Solution:
def minDays(self, bloomDay: List[int], m: int, k: int) -> int:
if m*k > len(bloomDay):
return -1
def define_ans(days:int) -> bool:
bouquets = flowers = 0
for i in range(len(bloomDay)):
if bloomDay[i] <= days:
flowers += 1
if flowers == k:
bouquets += 1
if bouquets == m:
break
flowers = 0
else:
flowers = 0
return bouquets == m
low,high = min(bloomDay),max(bloomDay)
while low < high:
days = (low + high) // 2
if define_ans(days):
high = days
else:
low = days + 1
return low