4.2每日一题 数列分段 【二分】

每日一题

####P1182 数列分段 Section II

题目描述

对于给定的一个长度为N的正整数数列 A_{1\sim N}A1∼N，现要将其分成 MM（M\leq NM≤N）段，并要求每段连续，且每段和的最大值最小。

关于最大值最小：

例如一数列4 2 4 5 1 要分成 3 段。

将其如下分段：

4 2\4 5\1

第一段和为6，第 22 段和为 9，第 33 段和为 1，和最大值为 9。

将其如下分段：

4\2 4\ 5 1

第一段和为 4，第 2 段和为 6，第 3 段和为 6，和最大值为 6。

并且无论如何分段，最大值不会小于 6。

所以可以得到要将数列 4 2 4 5 1 要分成 3 段，每段和的最大值最小为 6。

输入格式

第 11 行包含两个正整数 N,M。

第 22 行包含 N个空格隔开的非负整数 Ai，含义如题目所述。

输出格式

一个正整数，即每段和最大值最小为多少。

输入输出样例

输入 #1

5 3
4 2 4 5 1

输出 #1

6

说明/提示

对于 20% 的数据，N≤10。

对于 40% 的数据，N≤1000。

对于 100% 的数据，N≤105，M≤N，Ai < 10^8， 答案不超过 10^9。

题解

#include <bits/stdc++.h>
using namespace std;
int n,m;
int a[100005];
int f(int x){//判断是否每一段都小于mid
	int num = 1;
	int sum = 0;
	for(int i = 0;i < n;i++){
		if(sum + a[i] <= x)//若小于，添加
            sum += a[i];
		else{//否则另开一段
			sum = a[i];
			num++;
		}
	}
	if(num <= m) 
		return 1;
	return 0;
}
int main(){
	cin>> n >> m ;
	int l = 0,r = 0;
	int mid;
	for(int i = 0;i < n;i++){
		scanf("%d",&a[i]);
		l = max(l,a[i]);
		r += a[i];
	}
	while(l <= r){//二分
		mid = (l + r) / 2;
		if(f(mid))//mid大了，r左移
			r = mid - 1;
		else//小了
			l = mid + 1;
	}
	cout << l << endl;
	return 0;
}