考场上二分以后只想了贪心,没想DP
看见题解里的DP1秒想出来,卧槽巨水,完了水题也不会做了。
考虑一个答案,然后dp[i]表示分配前i本书最多可以分给几个人,然后如果无法以i结尾就不更新dp数组
然后把前缀和离散化,dp[i]=max(dp[j]+1),sum[i]-sum[j]<=mid ,那么就用sum[j]>=sum[i]-mid,然后二分出最小的离散化值,权值树状数组维护下标表示的sum[j]能得到的最大dp[j]。然后注意由于是要前i个完全分配,所以一定要有解的情况才更新dp[i]和权值树状数组
#include<bits/stdc++.h>
using namespace std;
#define maxl 200010
int n,tot,cnt,k;
long long up,dow,ans;
int a[maxl],b[maxl],dp[maxl],h[maxl],c[maxl];
long long sum[maxl];
long long num[maxl];
inline void prework()
{
scanf("%d%d",&n,&k);
up=0;dow=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
sum[i]=sum[i-1]+a[i],num[i]=sum[i];
if(a[i]>0) up+=a[i];
else dow+=a[i];
}
num[n+1]=0;
sort(num+1,num+2+n);
tot=unique(num+1,num+2+n)-num-1;
for(int i=0;i<=n;i++)
b[i]=lower_bound(num+1,num+1+tot,sum[i])-num;
}
inline void upd(int i,int x)
{
int t;
while(i<=tot && i)
{
h[i]=c[i];
for(int k=1;k<(i&-i);k<<=1)
h[i]=max(h[i],h[i-k]);
i+=i&-i;
}
}
inline int qry(int l,int r)
{
int ret=-n-1,len=r-l+1;
while(len && r)
{
if(len<(r&-r))
{
ret=max(ret,c[r]);
r--;len--;
}else
{
ret=max(ret,h[r]);
len-=(r&-r);
r-=r&-r;
}
}
return ret;
}
inline bool jug(long long mid)
{
for(int i=1;i<=n;i++)
dp[i]=0;
for(int i=1;i<=tot;i++)
h[i]=-n-1,c[i]=-n-1;
int l=1,r=n,id,tmp;
c[b[0]]=0;upd(b[0],c[b[0]]);
for(int i=1;i<=n;i++)
{
id=lower_bound(num+1,num+1+tot,sum[i]-mid)-num;
if(id>tot)
continue;
tmp=qry(id,tot);
if(tmp<=-n-1)
continue;
dp[i]=max(dp[i],qry(id,tot)+1);
c[b[i]]=max(c[b[i]],dp[i]);
upd(b[i],c[b[i]]);
if(dp[i]>=k)
return true;
}
return false;
}
inline void mainwork()
{
long long l=dow,r=up,mid;
while(l+1<r)
{
mid=(l+r)>>1;
if(jug(mid))
r=mid;
else
l=mid;
}
if(jug(l))
ans=l;
else
ans=l+1;
}
inline void print()
{
printf("%lld\n",ans);
}
int main()
{
int t;
scanf("%d",&t);
for(int i=1;i<=t;i++)
{
prework();
mainwork();
print();
}
return 0;
}