欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

洛谷P1182 数列分段 Section II 【贪心+二分】

程序员文章站 2022-07-16 10:51:32
...

题目链接:P1182 数列分段 Section II

洛谷P1182 数列分段 Section II 【贪心+二分】

程序说明:

二分适用于区间有界单调,例如最大值最小或者最小值最大这类问题。

设区间所有数中的最大值为L,区间所有的数总和为R,则答案一定位于L和R之间。利用二分算法快速得出答案。check函数利用贪心的思想,即P1181 数列分段Section I的方法,判断mid是否满足题目要求,如果不满足则在相应的区间进行二分查找。

代码如下:

#include <iostream>
using namespace std;
const int N = 100010;
int a[N], n, m; 
bool check(int x) {
	int tot = 0, cnt = 1;
	for(int i = 0; i < n; i++) {
		tot += a[i];
		if(tot > x) {
			cnt++;
			tot = a[i];
		}
	}
	//如果所分的段数大于m,返回true 
	return cnt > m;
}
int main() {
	cin>>n>>m;
	int l = 0, r = 0, mid;
	//l是区间最大值,r是区间总和 
	for(int i = 0 ; i < n; i++) {
		cin>>a[i];
		l = max(l, a[i]);
		r += a[i];
	}	
	while(l <= r) {
		mid = l + r >> 1;
		if(check(mid)) l = mid + 1;
		else r = mid - 1;
	}
	cout<<l;
	return 0;
}