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;
    }
};

还可以这样实现

class Solution {
public:
    int minDays(vector<int>& bloomDay, int m, int k) {
        if(m*k>bloomDay.size()) return -1;
        int l=0, r=*max_element(bloomDay.begin(), bloomDay.end()), mid;
        // cout<<l<<" "<<r<<endl;;
        // 这里为什么不能取等号呢?
        // 因为 若l=r而循环内的judge返回true时,死循环
        //若l=r且循环内的judge返回false时会不会错过答案呢?不会,因为此时l=mid+1>r,无法进入循环,跳出来返回r
        while(l<r){
            mid=l+(r-l)/2;
            if(judge(bloomDay, m, k, mid))r=mid;
            else l=mid+1;
            // cout<<l<<" "<<r<<" "<<mid<<endl;
        }
        return r;
    }
    bool judge(vector<int>& bloomDay,int m, int k, int mid){
        int cnt=0, window=0;
        for(int i=0;i<bloomDay.size();i++){
            if(bloomDay[i]<=mid) window++;
            else window=0;
            
            if(window==k){
                cnt++;
                window=0;
            }
        }
        // cout<<mid<<" "<<cnt<<endl;
        return cnt>=m;
    }
};

参考资料

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值