Editorial for Two

Editorial for Two

题意

给你n个数，找k个数出来，顺序为原来数组的顺序，分成两部分，求各部分之和的较大值的最小值。

思路

二分答案。判断是否能凑出两部分，选的数量之和大于等于k且两个部分分别的和都小于等于二分的答案，就可以缩小答案。二分答案妙啊。

实现

#include<iostream>
#include<queue>
using namespace std;
typedef long long ll;
const int maxn=3e5+5;
int t,n,k,a[maxn];
ll sum;//选的数之和
int l[maxn],r;//前i个数凑出小于等于二分ans最多能用的数量
priority_queue<int> q;//记录所选的数,贪心地选较小的数
//大根堆是为了pop大的数
bool check(ll ans){
	sum=0;
	while(!q.empty())
		q.pop();
	for(int i=1;i<=n;i++){//从前往后
		l[i]=l[i-1];
		if(sum+a[i]<=ans){//能装直接加
			q.push(a[i]);
			sum+=a[i];
			l[i]++;
		}
		else if(!q.empty()&&a[i]<q.top()){//装不下换出最大的
			sum-=q.top()-a[i];
			q.pop();
			q.push(a[i]);
		}//细节
	}
	sum=r=0;
	while(!q.empty())
		q.pop();
	for(int i=n;i>1;i--){//从后往前同理
		if(sum+a[i]<=ans){
			q.push(a[i]);
			sum+=a[i];
			r++;
		}
		else if(!q.empty()&&a[i]<q.top()){
			sum-=q.top()-a[i];
			q.pop();
			q.push(a[i]);
		}
		if(l[i-1]+r>=k)
			return 1;
	}
	return 0;
}
int main(){
	ll cl,cr,mid;
	scanf("%d",&t);
	while(t--){
		cl=1;cr=0;
		scanf("%d%d",&n,&k);
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
			cr+=a[i];
		}
		while(cl<cr){
			mid=cl+cr>>1;
			if(check(mid))
				cr=mid;
			else
				cl=mid+1;
		}
		printf("%lld\n",cr);
	}
}