（洛谷）【P1182】 数列分段 [高性能][贪心][二分]

查看原题

 

---

一道十分基础的二分算法题，却折磨了我好久（￣▽￣）

使和的最大值最小，这类题一看就是二分算法。同时，由于是连续分段，也就是说同一段里的数字，在数组中的排列必须是连续的，而不能在数组中随即找出几个数字进行排列，因此可以想到贪心算法。具体的对于二分、贪心的解释，请自行与百度搜索，仔细阅读文章并理解。

 

接下来放出程序。

#include<bits/stdc++.h>
#define cycle(i, n) for(int i=1; i<=n; i++) //高效替代for循环
using namespace std;
int m, n; //m sections, n number of numbers
int rgt = 0, lft = 0, mid = 0, ans = 0; 
int number[100005];

bool check(int section_size) { //检查函数
	int recent = 0, section_number = 1;
	cycle(i, n) {
		if(number[i]+recent<=section_size) { //贪心算法部分
			recent += number[i]; //够放，于是放东西
		}else {
			section_number++;
			recent = number[i]; //不够放，所以另开一段放东西
		}
	}
	return section_number <= m; //返回布朗型，以判断是否能在该情况下成功分段
}

int main() {
	//freopen("D:/C++ Library/p1182test.in","r",stdin);
	//freopen("D:/C++ Library/p1182test.out","w",stdout);
        //用于测试洛谷上天大的数据
	cin >> n >> m;
	cycle(i ,n) {
		cin >> number[i];
		if(number[i] > lft) lft = number[i];
		rgt += number[i];
	} //二分左边界为最大数，右边界为所有数字之和。这个很好理解
        //极限状态下，最大数可自分一段；另一种极限状态下，所有数全部分为一段
        //因此以这两个阈值的区间作为二分范围
	while(rgt>=lft) {
		mid = rgt + lft >> 1;
		if(check(mid)) {
			ans = mid;
			rgt = mid-1;
		}else {
			lft = mid+1;
		} 
	} //标准while二分方式，也可用另外的函数替代
	cout << ans;
	return 0;
}