蓝桥杯ACwing习题

https://www.acwing.com/problem/content/description/1030/

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 510 , INF = 0x3f3f3f3f;

int m , k;
int w[N] , L[N] , R[N];

bool check(int mid)
{
    // 我现在已经知道了这个最长的长度了
    // 我们需要知道的是每一个区间的范围 l 和 r
    // 设当前区间是 cnt 由于 mid 是最大的区间和 所以我们在枚举的时候如果大于 mid 的话 就表示当前区间的结束 
    // 同时记录一下当前区间的 左端点和右端点 
    // 因为我们是从后往前枚举的 , 所以每一次的 else 就是 一段区间的结束 cnt ++ , r[cnt] = i 
    // 那么上一个区间的左端点是 i + 1 因为 我们判断if 的时候 已经枚举到 i 了 但是算的 sum 没有包含 i 
    // 所以 l[cntlast] = i + 1 
    // 我们后面做的时候需要记得给最后一个区间的左端点 重新赋值 , 防止枚举到第一个 w[i] 的时候 sum 大了
    // 导致 单独有一个 w[1] 是没有 设置左端点的 那么左端点就自动补全为 0  在这个下标从1开始就错了
    // 1 1
    // 1
    // 上面的就是错误样例
    // 最后我们判断的是 区间的数量是否是 <= k 如果大于 k 的话 抽屉原理说明要满k个人足必定有一个人的抽屉里>= mid
    int s = w[m] , cnt = 1;
    R[cnt] = m;
    
    for(int i = m - 1 ; i >= 1 ; i --)
    {
        if(s + w[i] <= mid) s += w[i];
        else
        {
            L[cnt] = i + 1;
            s = w[i] , cnt ++;
            R[cnt] = i;
        }
    }
    L[cnt] = 1;
    if(cnt <= k) return true;
    return false;
}

int main()
{
    cin >> m >> k;
    
    int l = -INF , r = INF; // 枚举的是长度 所以 l 表示的是 
    for(int i = 1 ; i <= m ; i ++)
    {
        cin >> w[i];
        l = max(l , w[i]);
    }
    
    while(l < r)
    {
        int mid = l + r >> 1;
        if(check(mid)) r = mid;
        else l = mid + 1;
    }
    // 二分只是为了求出满足条件的长度 , 但题目是为了求区间 
    check(r); // 求出当前长度下的所有区间
    // 因为从后向前增加区间的 , 所以从后向前输出
    for(int i = k ; i >= 1 ; i --) cout << L[i] << ' ' << R[i] << endl ;
    
    return 0;
}