卡车的最小载重量(二分答案)

题目描述
码头上放置了n个集装箱(2<=n<=1000,000)每一个集装箱重量为 ai (1=<ai<=1000) ,码头空间很小，现在给你m辆同一个型号的卡车来完成装载任务，则卡车的最小载重量应该是多少，（货物必须按照原来的顺序被装载，卡车的载重重量必须<=最小载重量）
输入
集装箱个数n，和重量 ai 卡车数量

输出
最小载重量
样例输入

5
1 3 7 6 1
3

样例输出

7

提示
对于第一组测试数据共有5个货物1 3 7 6 1 需要3辆卡车 则可以让第一辆车装载1,3第二辆车装载7，第三辆车装在6，1

所以载重量为7的卡车就可以满足要求

解题思路

涉及到最小化最大值问题、最大化最小值问题，就要考虑能否通过二分答案求解。

题解.cpp

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

void solve() {
    int n;
    scanf("%d", &n);
    int a[n];
    LL l = 0, r = 0;
    for(int i = 0; i < n; i++) {
        scanf("%d", &a[i]);
        r += a[i];
    }
    int m;
    scanf("%d", &m);

    function <bool(LL)> judge = [&](LL w) {
        LL cnt = 0, sum = 0;
        for(int i = 0; i < n; i++) {
            if(a[i] > w) return false;
            if(sum + a[i] <= w) sum += a[i];
            else {
                cnt++;
                sum = a[i];
            }
        }
        return cnt < m ? true : false;
    };

    LL ans = r;
    while(l <= r) {
        LL mid = (l + r) >> 1;
        if(judge(mid)) {
            ans = min(ans, mid);
            r = mid - 1;
        }
        else l = mid + 1;
    }
    printf("%lld", ans);
}

int main() {
    solve();
    return 0;
}

题解.java

import java.util.Scanner;
public class Main {
    public static void main(String[] args) {
        Scanner cin = new Scanner(System.in);
        int n = cin.nextInt();
        int []a = new int[n+10];
        long l = 0,r = 0;
        for(int i=0;i<n;i++){
            a[i] = cin.nextInt();
            l = Math.max(a[i],l);
            r += a[i];
        }
        int m = cin.nextInt();
        long ans = 0;
        while(l <= r){
            long mid = (l + r) >> 1;
            long sum = 0,cnt = 0;
            for(int i=0;i<n;i++){
                if(mid < a[i]){
                    cnt = m+1;
                    break;
                }
                if(a[i] + sum <= mid){
                    sum += a[i];
                } else {
                    sum = a[i];
                    cnt++;
                }
            }
            if(cnt >= m){
                l = mid + 1;
            } else {
                r = mid - 1;
                ans = mid;
            }
        }
        System.out.println(ans);
    }
}