1436. 数列分段II

信息学奥赛一本通 二分查找

http://ybt.ssoier.cn:8088/problem_show.php?pid=1436

【题目描述】

对于给定的一个长度为N的正整数数列A[i]，现要将其分成M（M≤N）段，并要求每段连续，且每段和的最大值最小。

关于最大值最小：

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

将其如下分段：

[4 2][4 5][1]

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

将其如下分段：

[4][2 4][5 1]

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

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

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

【输入】

第1行包含两个正整数N，M，第2行包含N个空格隔开的非负整数A[i]，含义如题目所述。

【输出】

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

【输入样例】

5 3
4 2 4 5 1

【输出样例】

6

【分析】

本题可以用二分查找来解决，left等于序列中最大值，right等于序列和，在left与right之间查找出答案。

for(int i = 1; i <= n; i++){
		cin >> a[i];
		left = max(left, a[i]);
		right += a[i];
	}

取mid = （left+right）/ 2，计算子序列划分情况。

int mid = (left + right) / 2;
		int p = 1;//记载子序列数量，初始化为1
		int t = 0;//子序列和
		for(int i = 1; i <= n; i++){
			if(t + a[i] <= mid){
				t += a[i];
			}else{
				p++;//统计分段情况
				t = a[i];//新的一段开始
			}
		}

如果子序列划分p <=m，说明当前mid是一个有效的分割值，但可能还有更小的值也可以。此时，我们将 right 更新为 mid - 1，试图找到更小的值。

if(p <= m){
			right = mid - 1;
		}else{
			left = mid + 1;
		}

最终输出left的值，为什么是输出left，而不是right呢？因为最后一次迭代计算后，right被更新到一个不符合划分要求的值了，原题样例看不出来，现补充一个样例：

6 4
10 20 30 40 50 60

• 当第六次迭代时：

  • mid = (60 + 62) / 2 = 61
  • 分割子序列：最大和 ≤ 61
    • 10 + 20 = 30, 30 + 30 = 60, 40 = 100 > 61 (需要分割)
    • 子序列：[10, 20, 30], [40], [50], [60]
    • 子序列数 = 4，满足条件 (p = 4 ≤ 4)
  • 更新 right = mid - 1 = 60

• 结束条件：

  • left = 61
  • right = 60

最终，left 的值是 61，这是满足条件的最小的最大子序列和。而此时 right 已经是 60，但 left 是正确的答案。

【完整代码】

#include <bits/stdc++.h>
int a[100005];
using namespace std;
int main(int argc, char *argv[]) {
	int n,m;
	int left = 0, right = 0;
	cin >> n >> m;
	for(int i = 1; i <= n; i++){
		cin >> a[i];
		left = max(left, a[i]);//最大值
		right += a[i];//序列和
	}
	//二分查找
	while(left <= right){
		int mid = (left + right) / 2;
		int p = 1;//记载子序列数量，初始化为1
		int t = 0;//子序列和
		for(int i = 1; i <= n; i++){
			if(t + a[i] <= mid){
				t += a[i];
			}else{
				p++;//统计分段情况
				t = a[i];//新的一段开始
			}
		}
		//如果子序列数量p <= m，说明当前mid是一个有效的分割值，但可能还有更小的值也可以。此时，我们将 right 更新为 mid - 1，试图找到更小的值。
		if(p <= m){
			right = mid - 1;
		}else{
			left = mid + 1;
		}
	}
	cout << left;//输出
	return 0;
}