CF1560D Optimal Partition 题解

考场上想出来，但是打了一堆bug，手动捂脸一下

主要是一个naive的分段dp 然后对于val=0，为正，为负的可以用树状数组维护一下就好了，学到了一个新的维护后缀最大的方法，貌似把树状数组反着写一下就好了

#include <bits/stdc++.h>
using namespace std;
#define int long long
int tot,n,m;
int a[500100],f[500100],s[501000];
int b[500100];
int t[500100];
int t2[500100];
int tem[500100];
void add(int x,int val){
	while(x<=tot){
		t[x]=max(t[x],val);
		x+=x&-x;
	}
}
int s1(int x){
	int ans=-1e18;
	while(x){
		ans=max(t[x],ans);
		x-=x&-x;
	}return ans;
}
void add2(int x,int val){
	while(x){
		t2[x]=max(t2[x],val);
		x-=x&-x;
	}
}
int s2(int x){
	int ans=-1e18;
	while(x<=tot){
		ans=max(t2[x],ans);
		x+=x&-x;
	}return ans;
}
signed main()
{
int tt;cin>>tt;
while(tt--){
	cin>>n;s[0]=0;
	for(int i=1;i<=n;i++){
cin>>a[i];s[i]=s[i-1]+a[i];
f[i]=-1e18;b[i]=s[i];
}b[n+1]=0;sort(b+1,b+2+n);
 tot=unique(b+1,b+2+n)-b-1;
for(int i=1;i<=tot;i++)t[i]=t2[i]=tem[i]=-1e9;
for(int i=0;i<=n;i++)s[i]=lower_bound(b+1,b+1+tot,s[i])-b;
add(s[0],0);f[0]=0;
add2(s[0],0);//printf("pos=%lld ",tot);
tem[s[0]]=0;
for(int i=1;i<=n;i++){
/*	for(int j=1;j<=i;j++){
		int val;
		if(s[i]-s[j-1]>0)val=i-j+1;
		else if(s[i]==s[j-1])val=0;
		else val=-(i-j+1);
		f[i]=max(f[i],f[j-1]+val);
	}*/	
	f[i]=max(f[i],s1(s[i]-1)+i);
	f[i]=max(f[i],s2(s[i]+1)-i);
	f[i]=max(f[i],tem[s[i]]);//printf("%lld %lld %lld\n",i,b[i],s[i]);
	add(s[i],f[i]-(i));
	add2(s[i],f[i]+(i));
	tem[s[i]]=max(tem[s[i]],f[i]);
}for(int i=1;i<=n+1;i++)b[i]=0;
printf("%lld\n",f[n]);
}
}