强行二合一。。
前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);
}
}
}