划分数列

【问题描述】

给你一个有n个正整数元素的数列，要求把它划分成k段，使每段元素和的最大值最小。

【输入格式】sqea.in

输入第一行两个正整数n，k

第二行为此数列ai。

【输出格式】sqea.out

一个数，表示每段元素和的最大值最小的那个数。

【样例输入】

5 2

2 1 3 4 5

【样例输出】

9

【数据范围】

n <= 100000, k <= n, 0<=ai <= 109
talk is cheap，show me your code`

#include<iostream>
using namespace std;
const int maxx = 2e6 + 134;
long long a[maxx] = { 0 };
int n, k;
bool check(long long m) {//check  若m为段最大值需要划分几次。
	int temp = 0;
	int ctr = 1;
	for (int i = 1; i <= n; i++) {//遍历分段
		if (temp + a[i] < m) {//判断是否需要分段
			temp += a[i];
		}
		else {
			temp = a[i];
			ctr++;

		}
		
	}
	return ctr <= k;
}//如果分的段数大于了k，则说明这个m不合法
int main() {
	long long ls= maxx;
	long long rs = 0;
	cin >> n >> k;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		ls = min(ls, a[i]);
		rs += a[i];
	}
	while (ls + 1 < rs) {//如果m小于我们的目标答案，那么他会花费更多的k来分割数列，所以我们可以知道m若合法，则m+1会比m耗费更少的切割，所以m+1也是合法的
		long long mid = (ls + rs) >> 1;
		if (check(mid)) {
			rs = mid;
		}
		else {
			ls = mid;
		}
	}
	if (check(rs)) {
		cout << ls;
	}
	else {
		cout << rs;
	}
	
}