bzoj1926[Sdoi2010]粟粟的书架 主席树+前缀和

强行二合一。。

前50%首先想到200颗主席树,但好像空间会炸,然后看到h【i】<=1000,然后就可以暴力前缀和了

查询的时候二分,注意相等的要找到最左边,并特判无解

后50%直接线段树即可


码:

#include<iostream>
#include<cstdio>
using namespace std;
#define N 500005
int rt[N],sz[N*20],a[205][205],h[N],x,y,xx,yy,sum[N*20],ch[N*20][2],tot,c,d,b,n,m,q,i,j,k,qsum[1005][205][205],qsum2[1005][205][205],lie,lie2;
void up(int o)
{
	sz[o]=sz[ch[o][0]]+sz[ch[o][1]];
	sum[o]=sum[ch[o][0]]+sum[ch[o][1]];
}
void jia(int o,int l,int r,int last)
{
	if(b<=l&&r<=b)
	{
	sz[o]=sz[last]+1;
	sum[o]+=sum[last]+l;		
		return;
	}
	int mid=(l+r)>>1;
if(b<=mid)
{
ch[o][0]=++tot;
jia(ch[o][0],l,mid,ch[last][0]);
ch[o][1]=ch[last][1];	
}else
{
ch[o][1]=++tot;
jia(ch[o][1],mid+1,r,ch[last][1]);
ch[o][0]=ch[last][0];		
}
	up(o);		
}
void cha(int o,int last,int l,int r)
{
	if(l==r)
	{
		c+=d/l;
		if(d%l)c++;return;	
	}
	int mid=(l+r)>>1;
	if(sum[ch[o][1]]-sum[ch[last][1]]>=d)
	{
		cha(ch[o][1],ch[last][1],mid+1,r);	
	}else
	{
		d-=sum[ch[o][1]]-sum[ch[last][1]];
		c+=sz[ch[o][1]]-sz[ch[last][1]];
			cha(ch[o][0],ch[last][0],l,mid);	
	}
}
int main()
{
	scanf("%d%d%d",&n,&m,&q);
	if(n==1)
	{for(j=1;j<=m;j++)scanf("%d",&h[j]);
		for(i=1;i<=m;i++)
		{
		rt[i]=++tot;
		b=h[i];
		jia(rt[i],1,1000,rt[i-1]);					
		}
		while(q--)
		{c=0;
			scanf("%d%d%d%d%d",&x,&y,&xx,&yy,&d);
			if(d>sum[rt[yy]]-sum[rt[y-1]])
			{
				printf("Poor QLW\n");
			}else
			{
			cha(rt[yy],rt[y-1],1,1000);
			printf("%d\n",c);
		   }
		}
	}else
	{	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)scanf("%d",&a[i][j]);
		for(i=1;i<=1000;i++)
		{
			for(j=1;j<=n;j++)
			{
			lie=0;	lie2=0;			
			for(k=1;k<=m;k++)
			{
				if(a[j][k]>=i)lie+=a[j][k],lie2++;
				qsum[i][j][k]+=lie+qsum[i][j-1][k];
				qsum2[i][j][k]+=lie2+qsum2[i][j-1][k];
			}				
			}			
		}		
		while(q--)
		{
			c=0;
		scanf("%d%d%d%d%d",&x,&y,&xx,&yy,&d);
		int l=1,r=1001;int mid,o;
		while(l<r)
		{
			 mid=(l+r)>>1;
			  o=qsum[mid][xx][yy]+qsum[mid][x-1][y-1]-qsum[mid][x-1][yy]-qsum[mid][xx][y-1];
			if(o>d)
			l=mid+1;
			else r=mid;		
		}
	if(mid==1&&o<d){	printf("Poor QLW\n");continue;}
		d-=qsum[l][xx][yy]+qsum[l][x-1][y-1]-qsum[l][x-1][yy]-qsum[l][xx][y-1];	
		c+=qsum2[l][xx][yy]+qsum2[l][x-1][y-1]-qsum2[l][x-1][yy]-qsum2[l][xx][y-1];	
		l--;
		if(l)c+=d/l;
		if(l)if(d%l)c++;
		printf("%d\n",c);
		}		
	}
}



©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页