AC日记——[Sdoi2010]粟粟的书架 bzoj 1926

1926

思路：

　　主席树+二分水题；

代码：

#include <bits/stdc++.h>
using namespace std;
#define maxn 500005
#define maxr 205
#define maxn_ maxn*13
#define FalseAns "Poor QLW"
int n,m,q,lc[maxn_],rc[maxn_],num[maxn_],ci[maxn_],tot,root[maxn];
int ai[maxn],bi[maxn],vi[maxn],size,ansnum,map_[maxr][maxr];
int sum[maxr][maxr][],sumn[maxr][maxr][];
inline void in (int &now)
{
    char Cget=getchar();now=;
    while(Cget>''||Cget<'')Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}
void build(int &now,int l,int r)
{
    now=++tot;
    if(l==r) return;
    int mid=l+r>>;
    build(lc[now],l,mid);
    build(rc[now],mid+,r);
}
void add(int &now,int pre,int l,int r,int to,int x)
{
    now=++tot,num[tot]=num[pre]+,ci[tot]=ci[pre]+x;
    if(l==r) return;int mid=l+r>>;
    if(to<=mid) add(lc[now],lc[pre],l,mid,to,x),rc[now]=rc[pre];
    else add(rc[now],rc[pre],mid+,r,to,x),lc[now]=lc[pre];
}
void query(int now,int pre,int l,int r,int hi)
{
    if(l==r)
    {
        int pos=hi/bi[l];
        if(pos*bi[l]<hi) pos++;
        ansnum+=pos;
        return;
    }
    int mid=l+r>>,dis=ci[rc[now]]-ci[rc[pre]];
    if(hi<=dis) query(rc[now],rc[pre],mid+,r,hi);
    else
    {
        ansnum+=num[rc[now]]-num[rc[pre]];
        query(lc[now],lc[pre],l,mid,hi-dis);
    }
}
int main()
{
    in(n),in(m),in(q);
    if(n==)
    {
        for(int i=;i<=m;i++) in(ai[i]),bi[i]=ai[i],vi[i]=bi[i];
        sort(bi+,bi+m+),size=unique(bi+,bi+m+)-bi-;
        build(root[],,size);
        for(int i=;i<=m;i++)
        {
            ai[i]=lower_bound(bi+,bi+size+,ai[i])-bi;
            add(root[i],root[i-],,size,ai[i],vi[i]);
        }
        int x1,y1,x2,y2,hi,l,r;
        for(int i=;i<=q;i++)
        {
            in(x1),in(y1),in(x2),in(y2),in(hi);
            l=y1-,r=y2;
            if(ci[root[r]]-ci[root[l]]<hi) puts(FalseAns);
            else ansnum=,query(root[r],root[l],,size,hi),printf("%d\n",ansnum);
        }
    }
    else
    {
        for(int i=;i<=n;i++)
        {
            for(int j=;j<=m;j++)
            {
                in(map_[i][j]);
                for(int v=map_[i][j];v>=;v--)
                {
                    sum[i][j][v]++;
                    sumn[i][j][v]+=map_[i][j];
                }
                for(int v=;v<=;v++)
                {
                    sum[i][j][v]+=sum[i-][j][v]+sum[i][j-][v]-sum[i-][j-][v];
                    sumn[i][j][v]+=sumn[i-][j][v]+sumn[i][j-][v]-sumn[i-][j-][v];
                }
            }
        }
        int x1,y1,x2,y2,hi,l,r,mid,ans,x_,y_,x__,y__,x___,y___;
        int pos1,pos1_,pos2,pos2_;
        for(int i=;i<=q;i++)
        {
            in(x1),in(y1),in(x2),in(y2),in(hi);
            x_=x1-,y_=y2,x__=x2,y__=y1-,x___=x1-,y___=y1-;
            if(sumn[x2][y2][]-sumn[x_][y_][]-sumn[x__][y__][]+sumn[x___][y___][]<hi)
            {
                puts(FalseAns);
                continue;
            }
            l=,r=,ans=-;
            while(l<=r)
            {
                mid=l+r>>;
                if(sumn[x2][y2][mid]-sumn[x_][y_][mid]-sumn[x__][y__][mid]+sumn[x___][y___][mid]>=hi) ans=mid,l=mid+;
                else r=mid-;
            }
            pos1=sum[x2][y2][ans+]-sum[x_][y_][ans+]-sum[x__][y__][ans+]+sum[x___][y___][ans+];
            pos1_=sum[x2][y2][ans]-sum[x_][y_][ans]-sum[x__][y__][ans]+sum[x___][y___][ans];
            pos2=sumn[x2][y2][ans+]-sumn[x_][y_][ans+]-sumn[x__][y__][ans+]+sumn[x___][y___][ans+];
            pos2_=sumn[x2][y2][ans]-sumn[x_][y_][ans]-sumn[x__][y__][ans]+sumn[x___][y___][ans];
            hi-=pos2,pos2_-=pos2,pos2=hi/ans;if(pos2*ans<hi) pos2++;
            printf("%d\n",pos2+pos1);
        }
    }
    return ;
}