LeetCode LCP 12 - 小张刷题计划

题目描述

LCP 12. 小张刷题计划

解法：二分法（C++）

很清楚的是题意就是：给定一个数组，将其划分成 $M$ 份，使得每份元素之和最大值最小，每份可以任意减去其中一个元素。

如果不考虑每份可以任意减去一个元素，就是一个经典的二分问题。

但是本题加了一个条件：每份可以删除任意一个数。为了能够让最大值最小，显然每份中删去的一定是最大元素，所以在二分判定可行性时，维护当前序列的最大值，然后记录删除最大值的结果。

class Solution {
public:
    int minTime(vector<int>& time, int m) {
        if((int)time.size()<=m) return 0;
        int left = 0, right = INT_MAX;
        while(left<right)
        {
            int mid = left+(right-left)/2;
            if(canSplit(time, m, mid)) right = mid;
            else left = mid+1;
        }
        return right;
    }

    bool canSplit(vector<int>& time, int m, int split_num){
        int cnt = 0, sum = 0, maxtime = 0;
        for(auto& t: time)
        {
            sum += t;
            maxtime = max(maxtime, t); // 根据题意，可以删除子数组中最大的元素
            if(sum-maxtime>split_num)
            {
                if(++cnt==m) return false; // 划分了m次，有m+1个子数组
                sum= t;
                maxtime = t;
            }
        }
        return true;
    }
};