分组 CS精英挑战营 习题

分而治之

#include <bits/stdc++.h>
using namespace std;

// ================= 代码实现开始 =================

/* 请在这里定义你需要的全局变量 */

// 将所给数组分成连续的m份，使得数字之和最大的那一份的数字之和最小
// n：数组大小
// m：题中的m
// a：所给数组，大小为n
// 返回值：最优方案中，数字之和最大的那一份的数字之和
bool check(long long d, int n, int m, vector<int> &a){
    long long sum = 0; //用于记录当前那一份数字之和
    int cnt = 1;//在每一份数字不超过d的情况下，至少要分成的份数
    
    for(int i = 0; i<n; ++i){
        if(a[i]>d) //如果单个元素大于了d，直接返回false
            return false;
            
        sum += a[i]; //将a[i]加入到当前这一份中
        if(sum>d){
            sum=a[i];cnt++;//加入后超过d，需要重置sum
        }
        if(cnt>m)
            return false;
    }
    return true;
}


long long getAnswer(int n, int m, vector<int> a) {
    //l表示答案下界，r表示答案上界
    long long l=1,r=0;
    
    //求出初始x上界
    for(int i=0; i<n; i++)
        r += a[i];
    
    //二分答案
    while( l<=r ){
        long long mid = (l+r) >> 1;
        if(check(mid,n,m,a))//验证mid是否可能为答案如果可能上界缩小
            r = mid-1;
        else // 否则下界增大
            l = mid+1;
    }
    return r+1; //答案为r+1 因为r+1是最后一个满足check的mid值
}

// ================= 代码实现结束 =================

int main() {
    int n, m;
    scanf("%d%d", &n, &m);
    vector<int> a;
    a.resize(n);
    for (int i = 0; i < n; ++i)
        scanf("%d", &a[i]);
    printf("%lld\n", getAnswer(n, m, a));
    return 0;
}