Leetcode 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

解題報告

很明顯，用二分法。

實現代碼

class Solution {
public:
    int minDays(vector<int>& bloomDay, int m, int k) {
        if(m*k>bloomDay.size()) return -1;
        int n=bloomDay.size();
        
        int l=0,r=*max_element(bloomDay.begin(), bloomDay.end()), mid;
        while(l<=r){
            mid=l+(r-l)/2;
            int result =judge(mid, m, k, bloomDay);
            if(result) r=mid-1;
            else l=mid+1;
        }
        return l;
    }
    int judge(int target, int m, int k, vector<int>& bloomDay){
        int counter=0, window=0;
        for(int i=0;i<bloomDay.size();i++){
            
            if(bloomDay[i]<=target) window++;
            else window=0;
            
            if(window==k){
                window=0;
                counter++;
            }
        }
        return counter>=m;
    }
};

參考資料

[1] Leetcode 1482. 製作m束花所需的最少天數