数列分段 II（二分 + 贪心）

最新推荐文章于 2024-05-17 13:26:14 发布

念兮兮

最新推荐文章于 2024-05-17 13:26:14 发布

阅读量56

点赞数

分类专栏：《信息学奥赛一本通-提高篇》在线题库文章标签： c++ 算法数据结构

本文链接：https://blog.csdn.net/qq_52081267/article/details/129694866

版权

《信息学奥赛一本通-提高篇》在线题库专栏收录该内容

40 篇文章 0 订阅

订阅专栏

原题链接

解题思路：先二分答案，再模拟判断是否满足题目要求，当l == r时，r即为答案

C++代码

#include <bits/stdc++.h>
#define int long long
#define LL long long
using namespace std;
const int N = 1e5 + 10;
int n, m;
int a[N];

bool check(int mid)
{
    int sum = 0, cnt = 0;
    for (int i = 1; i <= n; i ++ )
    {
        sum += a[i];
        if (sum >= mid)
        {
            cnt ++ ;
            if (sum == mid) sum = 0;
            else sum = a[i];
        }
    }
    if (sum) cnt ++ ;
    return cnt <= m;
}

signed main()
{
    cin >> n >> m;
    int l = 0, r = 0;
    for (int i = 1; i <= n; i ++ ) 
    {
        cin >> a[i];
        l = max(l, a[i]);
        r += a[i];
    }
    while (l < r)
    {
        int mid = l + r >> 1;
        if (check(mid)) r = mid;
        else l = mid + 1;
    }
    cout << r << endl;
    return 0;
}