bzoj1926: [Sdoi2010]粟粟的书架

传送门
对于第一问(R,C<=200)；
预处理f[x][y][k],s[x][y][k].表示从(1,1)到(x,y)中大于等于k的数的和与大于等于k的数的个数。
然后二分最小的数即可。
对于第二问(R=1):
我们还是二分最小数。
判断就变成了询问一段区间内大于等于x的数的和以及它们的个数。
显然主席树可以处理这个

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<queue>
#define N 11000000
#define M 210 
using namespace std;
int n,m,T,a[M][M][M*5],b[M][M][M*5],c[M][M],x1,y1,x2,y2,h;
int t,rt[550000],ls[N],rs[N],v[N],s[N],cnt;
void work1(){
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++) scanf("%d",&c[i][j]);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            for (int k=1;k<=1000;k++){
                a[i][j][k]=a[i][j-1][k]+a[i-1][j][k]-a[i-1][j-1][k];
                b[i][j][k]=b[i][j-1][k]+b[i-1][j][k]-b[i-1][j-1][k];
                if (c[i][j]>=k) a[i][j][k]++,b[i][j][k]+=c[i][j];
            }
    while (T--){
        scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&h);
        int l=0,r=1001; x1--; y1--;
        while (l+1<r){
            int mid=(l+r)/2;
            if (b[x1][y1][mid]+b[x2][y2][mid]-b[x1][y2][mid]-b[x2][y1][mid]>=h)
                l=mid; else r=mid;
        }
        if (l==0) puts("Poor QLW");
        else{
            int sum1=b[x1][y1][l]+b[x2][y2][l]-b[x2][y1][l]-b[x1][y2][l]-h;
            int sum2=a[x1][y1][l]+a[x2][y2][l]-a[x2][y1][l]-a[x1][y2][l];
            printf("%d\n",sum2-sum1/l);
        }
    }
}
void add(int l,int r,int x,int &y,int k){
    y=++cnt; v[y]=v[x]+1; s[y]=s[x]+k;
    if (l==r) return;
    int mid=(l+r)/2;
    if (k<=mid) rs[y]=rs[x],add(l,mid,ls[x],ls[y],k);
    else ls[y]=ls[x],add(mid+1,r,rs[x],rs[y],k);
}
int ask(int x,int y,int k){
    int l=1,r=1000,ok=0;
    x=rt[x-1]; y=rt[y];
    if (s[y]-s[x]<h) return -1;
    while (l<r){
        int mid=(l+r)/2,t=s[rs[y]]-s[rs[x]];
        if (t<k){
            ok+=v[rs[y]]-v[rs[x]]; k-=t; r=mid;
            x=ls[x]; y=ls[y];
        }
        else{
            l=mid+1; x=rs[x]; y=rs[y];
        }
    }
    ok+=(k+l-1)/l;
    return ok;
}
void work2(){
    for (int i=1,x;i<=m;i++){
        scanf("%d",&x);
        add(1,1000,rt[i-1],rt[i],x);
    }
    while (T--){
        scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&h);
        int ans=ask(y1,y2,h);
        if (ans==-1) puts("Poor QLW");
        else printf("%d\n",ans);
    }
}
int main(){
    scanf("%d%d%d",&n,&m,&T);
    if (n!=1) work1(); else work2();
}