数列分段
题目链接:ybt高效进阶1-3-1 \ luogu P1182
题目大意
就是要把一个序列分成 M 段,要每段和的最大值最小。
问你这个最小值可以到多小。
思路
这道题,我们就直接二分这个最大值,然后扫过这个数组,判断能不能分成
M
M
M 段或更短。
判断就是贪心一下,从第一个数开始扫过去,如果这一段加了这个值就超过了最大值,就让这个值放在另一段里;如果加了还不超过最大值,肯定是贪心让它放进来。
代码
#include<cstdio>
#include<iostream>
using namespace std;
int n, m, a[100001], l, r, mid, ans, sum, nt;
bool check(int mid) {
sum = 0;
nt = 0;
for (int i = 1; i <= n; i++) {
sum += a[i];
if (sum > mid) {
nt++;
if (nt > m) return 0;
sum = a[i];
}
}
if (sum) nt++;
if (nt <= m) return 1;
return 0;
}
int main() {
scanf("%d %d", &n, &m);
for (int i = 1; i <= n; i++) {
scanf("%d", &a[i]);
l = max(a[i], l);
r += a[i];
}
while (l <= r) {
mid = (l + r) >> 1;
if (check(mid)) {
ans = mid;
r = mid - 1;
}
else l = mid + 1;
}
printf("%d", ans);
return 0;
}