【BZOJ-1926】粟粟的书架 二分 + 前缀和 + 主席树

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
int read()
{
    int x=0,f=1; char ch=getchar();
    while (ch<‘0‘ || ch>‘9‘) {if (ch==‘-‘)f=-1; ch=getchar();}
    while (ch>=‘0‘ && ch<=‘9‘) {x=x*10+ch-‘0‘; ch=getchar();}
    return x*f; 
}
int R,C,M;
#define maxn 500010
int Sum[maxn*20],Num[maxn*20],ll[maxn*20],rr[maxn*20],root[maxn<<2],sz;
int sum[210][210][1010],num[210][210][1010],p[210][210];
void Insert(int l,int r,int &now,int fat,int val)
{
    now=++sz; Sum[now]=Sum[fat]+val; Num[now]=Num[fat]+1;
    if (l==r) return;
    ll[now]=ll[fat],rr[now]=rr[fat];
    int mid=(l+r)>>1;
    if (val<=mid) Insert(l,mid,ll[now],ll[fat],val);
    else Insert(mid+1,r,rr[now],rr[fat],val);
}
int Query(int l,int r,int L,int R,int kth)
{
    if (Sum[root[R]]-Sum[root[L-1]]<kth) return -1;
    L=root[L-1]; R=root[R];
    int re=0;
    while (l<r)
        {
            int mid=(l+r)>>1,tmp=Sum[rr[R]]-Sum[rr[L]];
            if (tmp<kth) {re+=Num[rr[R]]-Num[rr[L]]; kth-=tmp; r=mid; L=ll[L]; R=ll[R];}
                else {l=mid+1; L=rr[L]; R=rr[R];}
        }
    re+=(kth+l-1)/l;
    return re;
}
void Part1()
{
    for (int i=1; i<=C; i++) Insert(1,1000,root[i],root[i-1],read());
    while (M--)
        {
            int x1=read(),y1=read(),x2=read(),y2=read(),h=read();
            int ans=Query(1,1000,y1,y2,h);
            if (ans==-1) {puts("Poor QLW"); continue;}
            printf("%d\n",ans);
        }
}
int GetSum(int x1,int y1,int x2,int y2,int k) {return sum[x1-1][y1-1][k]+sum[x2][y2][k]-sum[x1-1][y2][k]-sum[x2][y1-1][k];}
int GetNum(int x1,int y1,int x2,int y2,int k) {return num[x1-1][y1-1][k]+num[x2][y2][k]-num[x1-1][y2][k]-num[x2][y1-1][k];}
void Part2()
{
    int maxx=0;
    for (int i=1; i<=R; i++)
        for (int j=1; j<=C; j++)
            p[i][j]=read(),maxx=max(maxx,p[i][j]);
    for (int i=1; i<=R; i++)
        for (int j=1; j<=C; j++)
            for (int k=1; k<=maxx; k++)
                num[i][j][k]=num[i-1][j][k]+num[i][j-1][k]-num[i-1][j-1][k]+(p[i][j]>=k?1:0),
                sum[i][j][k]=sum[i-1][j][k]+sum[i][j-1][k]-sum[i-1][j-1][k]+(p[i][j]>=k?p[i][j]:0);
    while (M--)
        {
            int x1=read(),y1=read(),x2=read(),y2=read(),h=read();
            int l=0,r=maxx+1,k=-1;
            while (l+1<r)
                {
                    int mid=(l+r)>>1;
                    if (GetSum(x1,y1,x2,y2,mid)>=h) l=mid,k=mid; else r=mid;
                }
            if (k==-1) {puts("Poor QLW"); continue;}
            printf("%d\n",GetNum(x1,y1,x2,y2,k)-(GetSum(x1,y1,x2,y2,k)-h)/k);
        }
}
int main()
{
    R=read(),C=read(),M=read();
    if (R==1) Part1(); else Part2();
    return 0;
}