HDU6606 Distribution of books 2019杭电多校第三场

考场上二分以后只想了贪心,没想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
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>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;
}

 

你可能感兴趣的:(树状数组,二分)