LeetCode-Python-1482. 制做 m 束花所需的最少天数(二分试探)

给你一个整数数组 bloomDay,以及两个整数 m 和 k 。python

现须要制做 m 束花。制做花束时,须要使用花园中 相邻的 k 朵花 。数组

花园中有 n 朵花,第 i 朵花会在 bloomDay[i] 时怒放,刚好 能够用于 一束 花中。网络

请你返回从花园中摘 m 束花须要等待的最少的天数。若是不能摘到 m 束花则返回 -1 。code

示例 1:leetcode

输入: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:io

输入:bloomDay = [1,10,3,10,2], m = 3, k = 2
输出:-1
解释:要制做 3 束花,每束须要 2 朵花,也就是一共须要 6 朵花。而花园中只有 5 朵花,没法知足制做要求,返回 -1 。
示例 3:class

输入: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:object

输入:bloomDay = [1000000000,1000000000], m = 1, k = 1
输出:1000000000
解释:须要等 1000000000 天才能采到花来制做花束
示例 5:List

输入:bloomDay = [1,10,2,9,3,8,4,7,5,6], m = 4, k = 2
输出:9
 im

提示:

bloomDay.length == n
1 <= n <= 10^5
1 <= bloomDay[i] <= 10^9
1 <= m <= 10^6
1 <= k <= n

来源:力扣(LeetCode)
连接:https://leetcode-cn.com/problems/minimum-number-of-days-to-make-m-bouquets
著做权归领扣网络全部。商业转载请联系官方受权,非商业转载请注明出处。

思路:

首先排除掉不可能的状况。

而后答案的区间是已知的,其必在 【1, max(bloomDay)】的区间里, 所以是典型的二分试探的题目。

因此能够利用二分法逐步缩小区间范围,找到最后答案。

时间复杂度:O(Nlog(max(bloomDay)))

空间复杂度:O(1)

class Solution(object):
    def minDays(self, bloomDay, m, k):
        """
        :type bloomDay: List[int]
        :type m: int
        :type k: int
        :rtype: int
        """
        if m * k > len(bloomDay): # 全部花开了都不行,无解
            return -1
        
        left, right = 1, max(bloomDay)
        while left <= right:
            mid = (left + right) // 2 # 本轮猜想的答案
             
            flowerCnt = 0
            dayCnt = 0
            for day in bloomDay:
                if day <= mid:
                    dayCnt += 1
                    if dayCnt == k: # 能够作一束花
                        flowerCnt += 1
                        dayCnt = 0
                else:
                    dayCnt = 0
            
            if flowerCnt >= m: # 当前mid可行,试试更少的天数行不行
                right = mid - 1
            else:
                left = mid + 1
                
        return left