【BZOJ】1926: [Sdoi2010]粟粟的书架（暴力+主席树）

题目

传送门：QWQ

分析

两道题目

第一问暴力预处理

用$ a[i][j][k] $和$ s[i][j][k] $ 表示从$ (1,1) $ 到 $ (i,j) $ 这个矩形中比k大的数的个数和这些数的和。

对于询问我们二分k，得出答案（最后还要减掉一些值一样的东西 详见代码）

第二问主席树

r=1时只有一个数列，我们要求的是用多少数可以大于等于h

用主席树搞，询问时就像询问区间第k大时一样向左向右走，只不过要再套个二分。

代码

#include <bits/stdc++.h>
using namespace std;
const int maxn=;
int s[][][],a[][][],p[][];
  int rt[maxn], newp, ls[], rs[], sum[], v[];
//int rt[maxn], newp, ls[10000100], rs[10000100], sum[10000100], v[10000100];

int r,c,m;
inline int in()
{
    char c=getchar();int x=;
    while (c<''||c>'') c=getchar();
    while (c>=''&&c<='') x=x*+c-'',c=getchar();
    return x;
}
void insert(int l,int r,int x,int &cur,int cur1){
    cur=++newp;
    ls[cur]=ls[cur1]; rs[cur]=rs[cur1];
    sum[cur]=sum[cur1]+x; v[cur]=v[cur1]+;
    if(l==r) return;
    int mid=l+r>>;
    if(x<=mid) /*ls[cur]=ls[cur1],*/insert(l,mid,x,ls[cur],ls[cur1]);
    else /*rs[cur]=rs[cur1],*/insert(mid+,r,x,rs[cur],rs[cur1]);
}
int query(int x,int y,int h)
{
    x=rt[x-];y=rt[y];
    if(sum[y]-sum[x]<h) return -;
    int l=,r=,ans=;
    while(l<r){
        int mid=(l+r)>>;
        if(sum[rs[y]]-sum[rs[x]]<h){
            ans+=v[rs[y]]-v[rs[x]]; h-=sum[rs[y]]-sum[rs[x]]; x=ls[x]; y=ls[y]; r=mid;
        //    printf("-----  %d\n",v[rs[y]]-v[rs[x]]);
        }
        else {
            l=mid+; x=rs[x]; y=rs[y];
        }
    }
    ans+=(h+l-)/l;
    return ans;
}
int main()
{
    r=in();c=in();m=in();
    if(r!=){
        for(int i=;i<=r;i++)
        for(int j=;j<=c;j++) p[i][j]=in();
        for(int i=;i<=r;i++)
        for(int j=;j<=c;j++)
        for(int k=;k<=;k++)
        {
            a[i][j][k]=a[i-][j][k]+a[i][j-][k]-a[i-][j-][k];
            s[i][j][k]=s[i-][j][k]+s[i][j-][k]-s[i-][j-][k];
            if(p[i][j]>=k){
                a[i][j][k]++; s[i][j][k]+=p[i][j];
            }
        }
        while(m--){
            int x1=in(),y1=in(),x2=in(),y2=in(),h=in();
            int l=,r=,ans; x1--;y1--;
            while(l+<r){
                int mid=(l+r)>>;ans=s[x2][y2][mid]-s[x2][y1][mid]-s[x1][y2][mid]+s[x1][y1][mid];
                if(ans>=h) l=mid; else r=mid;
            }
            int sum1=a[x2][y2][l]-a[x2][y1][l]-a[x1][y2][l]+a[x1][y1][l],
                sum2=s[x2][y2][l]-s[x2][y1][l]-s[x1][y2][l]+s[x1][y1][l];
            if(l==){puts("Poor QLW"); continue;}
            else printf("%d\n",sum1-(sum2-h)/l);
        }
    }
    else{
        for(int i=;i<=c;i++){
            int t=in();
            insert(,,t,rt[i],rt[i-]);
        }
        while(m--){
            int x1=in(),y1=in(),x2=in(),y2=in(),h=in();
            int ans=query(y1,y2,h);
            if(ans==-) puts("Poor QLW");
            else printf("%d\n",ans);
        }
    }
    return ;
}