显然可以二分。
然后二分之后,我们就可以dp了,dp[i]=max{dp[j]+1},保证j+1到i的和小于等于mid即可。
然后这个显然是区间max,可以用线段树优化,这道题很恶心,卡空间了,所以我们必须离散化。
AC代码:
#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include
#define int long long
using namespace std;
const int N=2e5+10,M=N*4;
int n,k,a[N],mx[M],dp[N],lc[M],rc[M],rt,cnt,up;
vector<int> v;
#define mid (l+r>>1)
void change(int &p,int l,int r,int x,int v){
if(!p) p=++cnt,mx[p]=-1e18,lc[p]=rc[p]=0;
if(l==r){mx[p]=max(mx[p],v); return ;}
if(x<=mid) change(lc[p],l,mid,x,v);
else change(rc[p],mid+1,r,x,v);
mx[p]=max(mx[lc[p]],mx[rc[p]]);
}
int ask(int p,int l,int r,int ql,int qr){
if(!p||ql>qr) return -1e18;
if(l==ql&&r==qr) return mx[p];
if(qr<=mid) return ask(lc[p],l,mid,ql,qr);
else if(ql>mid) return ask(rc[p],mid+1,r,ql,qr);
else return max(ask(lc[p],l,mid,ql,mid),ask(rc[p],mid+1,r,mid+1,qr));
}
#undef mid
inline int get(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}
inline int check(int mid){
rt=cnt=0; change(rt,1,up,get(0),0);
for(int i=1;i<=n;i++){
dp[i]=ask(rt,1,up,get(v[a[i]-1]-mid),up)+1;
change(rt,1,up,a[i],dp[i]);
if(dp[i]>=k) return 1;
}
return 0;
}
char *fs,*ft,buf[1<<20];
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read(){
int x=0,f=1; char ch=gc();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=gc();}
return x*f;
}
inline void solve(){
n=read(),k=read(); mx[0]=-1e18;
for(int i=1;i<=n;i++) a[i]=read(),a[i]+=a[i-1],v.push_back(a[i]); v.push_back(0);
sort(v.begin(),v.end()); v.erase(unique(v.begin(),v.end()),v.end()); up=v.size();
for(int i=1;i<=n;i++) a[i]=get(a[i]);
int l=-2e14,r=2e14;
while(l<r){
int mid=l+r>>1;
if(check(mid)) r=mid;
else l=mid+1;
}
cout<<l<<'\n';
}
signed main(){
int T; T=read(); while(T--) solve();
return 0;
}