考场上想出来,但是打了一堆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]);
}
}