给你一个整数数组 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 。
解题思路
每尝试一个时间,都遍历一遍bloomDay数组,找到符合要求的花束。
尝试时间的选择在bloomDay最小值和最大值这个区间进行二分选择。
class Solution(object):
def minDays(self, bloomDay, m, k):
"""
:type bloomDay: List[int]
:type m: int
:type k: int
:rtype: int
"""
n = len(bloomDay)
# 先判断特殊情况
if m*k>n:
return -1
elif m*k==n:
return max(bloomDay)
else:
# 定义一个函数判断当天数为day时能获取多少束花
def count(day):
# total为当天数为day时能获取的花的个数
total = 0
i = 0
cnt = 0
while i<n:
# 判断连续的满足条件的花的朵数cnt,当不满足条件之后跳出循环
while i<n and bloomDay[i]<=day:
cnt += 1
i += 1
# 计算cnt朵连续的花能合成的花束
total += cnt // k
cnt = 0
i += 1
return total>=m
# 利用二分法确定天数
left, right = min(bloomDay), max(bloomDay)
while left<right:
mid = (left + right) // 2
if count(mid):
right = mid
else:
left = mid + 1
return left