2023牛客OI赛前集训营-提高组（第三场） 分糖果

时间复杂度为$O(Tn\log n\log v)$。

code

#include
using namespace std;
int T,n,K,tr[200005],f[100005];
long long ans,a[100005],sum[100005];
vector<long long>v;
int lb(int i){
	return i&(-i);
}
void pt(int i,int w){
	while(i<=v.size()){
		tr[i]=max(tr[i],w);
		i+=lb(i);
	}
}
int find(int i){
	int re=-1e9;
	while(i){
		re=max(re,tr[i]);
		i-=lb(i);
	}
	return re;
}
int gtid(long long w){
	int t=lower_bound(v.begin(),v.end(),w)-v.begin();
	return v.size()-t;
}
bool pd(long long mid){
	v.clear();
	for(int i=1;i<=n;i++){
		v.push_back(sum[i]);
		v.push_back(sum[i]-mid);
	}
	v.push_back(0);
	sort(v.begin(),v.end());
	v.erase(unique(v.begin(),v.end()),v.end());
	for(int i=1;i<=v.size();i++) tr[i]=-1e9;
	f[0]=0;
	pt(gtid(0),f[0]);
	for(int i=1;i<=n;i++){
		f[i]=find(gtid(sum[i]-mid))+1;
		pt(gtid(sum[i]),f[i]);
		if(f[i]>=K) return 1;
	}
	return 0;
}
int main()
{
	scanf("%d",&T);
	while(T--){
		scanf("%d%d",&n,&K);
		for(int i=1;i<=n;i++){
			scanf("%lld",&a[i]);
			sum[i]=sum[i-1]+a[i];
		}
		long long l=-1e15,r=1e15,mid;
		while(l<=r){
			mid=l+r>>1;
			if(pd(mid)) r=mid-1;
			else l=mid+1;
		}
		printf("%lld\n",r+1);
	}
	return 0;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64