蓝桥杯—数列分段—二分答案

1、问题描述

题目描述

对于给定的一个长度为N的正整数数列 A​1~AN，现要将其分成 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 个空格隔开的非负整数 Ai​，含义如题目所述。

输出格式

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

输入输出样例

输入 

5 3
4 2 4 5 1

输出 

6

说明/提示

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

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

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

2、代码实现

import java.util.Scanner;

public class 数列分段 {
	static int n;
	static int m;
	static int[] arr;
	
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		n = scanner.nextInt();
		m = scanner.nextInt();
		arr = new int[n];
		int max = 0;
		int sum = 0;
		for(int i = 0;i < n;i++) {
			arr[i] = scanner.nextInt();
			max = Math.max(arr[i], max);
			sum += arr[i];
		}
		int left = max;
		int right = sum;
		int middle = 0;
		while(left < right) {
			middle = (left + right) >> 1;
			if(check(middle)) {
				right = middle;
			}else {
				left = middle + 1;
			}
		}
		System.out.println(left);
	}
	public static boolean check(int middle) {
		//当每段的值为middle时，我们需要分成m段
		int part = 1;
		int cur = 0;
		for(int i = 0;i < n;i++) {
			cur += arr[i];
			if(cur > middle) {
				cur = arr[i];
				part++;
			}
		}
		return part <= m;
	}
}