2019牛客暑期多校训练营(第九场) H Cutting Bamboos

题目描述:

英语差

题目分析:

思路来自这位大佬Orz
我们二分一个最高的整数,用主席树查询,满足区间内 ans及以上高度的和大于等于第x次切除的,把ans+1上的去处,然后就可以算出答案来了。
主席树上查树上的数字和以及数量

代码:


#include 
#include 
#include 
#define int long long
const int maxm=2e5+10;
const int N=100000;
int sum[maxm],h[maxm],root[maxm];
int n,m;
struct tree{
	int sz;
	int sum[maxm*40],cnt[maxm*40];
	int lson[maxm*40],rson[maxm*40];
	int anssum,anscnt;
	void clear(){anssum=0,anscnt=0;}
	void ins(int &now,int pre,int l,int r,int ind)
	{
		now=++sz;
		lson[now]=lson[pre],rson[now]=rson[pre];
		sum[now]=sum[pre]+ind,cnt[now]=cnt[pre]+1;
		if(l>=r) return;
		int mid=(l+r)>>1;
		ind<=mid?ins(lson[now],lson[pre],l,mid,ind):ins(rson[now],rson[pre],mid+1,r,ind);
	}
	void ask(int now,int pre,int l,int r,int ind)
	{
		if(ind<=l)
		{
			anssum+=sum[now]-sum[pre];
			anscnt+=cnt[now]-cnt[pre];
			return;
		}
		int mid=(l+r)>>1;
		if(ind<=mid) ask(lson[now],lson[pre],l,mid,ind);
		ask(rson[now],rson[pre],mid+1,r,ind);
	}
}mtree;
signed main()
{
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&h[i]);
		sum[i]=sum[i-1]+h[i];
		mtree.ins(root[i],root[i-1],1,N,h[i]);
	}
	for(int i=1;i<=m;i++)
	{
		int ql,qr,x,y;
		scanf("%lld%lld%lld%lld",&ql,&qr,&x,&y);
		double tmp,hi=1.0*(sum[qr]-sum[ql-1])*x/y;
		int l=1,r=N;
		double ans=0;
		while(l<=r)
		{
			mtree.clear();
			int mid=(l+r)>>1;
			mtree.ask(root[qr],root[ql-1],1,N,mid);
			if(mtree.anssum-mtree.anscnt*(mid-1)>=hi)
			{
				ans=mid;
				tmp=1.0*mtree.anscnt;
				l=mid+1;
			}
			else r=mid-1;
		}
		double Ans;
		if(ans!=N)
		{
			mtree.clear();
		    mtree.ask(root[qr],root[ql-1],1,N,ans+1);
		    Ans=mtree.anssum-mtree.anscnt*ans;
		}
		else Ans=0.0;
		printf("%.15lf\n",ans-(hi-Ans)/tmp);
	}
	return 0;
}

你可能感兴趣的:(题目分析,主席树,二分)