1926: [Sdoi2010]粟粟的书架|主席树|二分答案

ym_fqk神犇提供做法

一边看AlphaGo VS 李世石一边写这个sb题居然1A辣。
做法显然。n=1时就是序列上主席树然后二分答案，否则就是区间维护1000个前缀和暴力再二分答案。最后要暴力一下最小值的出现次数。

暴力出奇迹，内存贴线过，差点被卡飞。。

#include<set>
#include<map>
#include<ctime>
#include<queue>
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
int sc()
{
    int i=0; char c=getchar();
    while(c>'9'||c<'0')c=getchar();
    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
    return i;
}
int sum[1005][205][205],s[1005][205][205];
int a[205][205],b[500005],root[500005];
int t[8000008],sm[8000008],ch[8000008][2];
int n,m,Q,cnt;
int SUM(int d,int x1,int y1,int x2,int y2)
{
    return sum[d][x2][y2]-sum[d][x1-1][y2]-sum[d][x2][y1-1]+sum[d][x1-1][y1-1];
}
void insert(int pre,int &x,int l,int r,int v)
{
    if(!x)x=++cnt;
    sm[x]=sm[pre]+v;t[x]=t[pre]+1;
    if(l==r)return;
    int mid=l+r>>1;
    if(v<=mid)
        ch[x][1]=ch[pre][1],
        insert(ch[pre][0],ch[x][0],l,mid,v);
    else 
        ch[x][0]=ch[pre][0],
        insert(ch[pre][1],ch[x][1],mid+1,r,v);
}
int query(int L,int R,int H)
{
    if(sm[R]-sm[L]<H)return 0;
    int l=1,r=1000,tot=0;
    while(l!=r)
    {
        int mid=l+r>>1;
        int re=t[ch[R][1]]-t[ch[L][1]];
        int res=sm[ch[R][1]]-sm[ch[L][1]];
        if(res>=H)
            L=ch[L][1],R=ch[R][1],l=mid+1;
        else L=ch[L][0],R=ch[R][0],r=mid,H-=res,tot+=re;
    }
    tot+=t[R]-t[L];
    int ans=sm[R]-sm[L];
    while(ans-l>=H)ans-=l,tot--;
    return tot;
}
void solve1()
{
    for(int i=1;i<=m;i++)
        b[i]=sc(),insert(root[i-1],root[i],1,1000,b[i]);
    while(Q--)
    {
        int x1=sc(),l=sc(),x2=sc(),r=sc(),H=sc();
        int ans=query(root[l-1],root[r],H);
        if(!ans) puts("Poor QLW");
        else printf("%d\n",ans);
    }
}
void solve2()
{
    int mx=0;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            mx=max(mx,a[i][j]=sc());
    for(int i=1;i<=mx;i++)
        for(int j=1;j<=n;j++)
            for(int k=1;k<=m;k++)
            {
                sum[i][j][k]=sum[i][j-1][k]+sum[i][j][k-1]-sum[i][j-1][k-1];
                s[i][j][k]=s[i][j-1][k]+s[i][j][k-1]-s[i][j-1][k-1];
                if(a[j][k]==i)sum[i][j][k]+=i,s[i][j][k]++;
            }
    for(int i=mx-1;i;i--)
        for(int j=1;j<=n;j++)
            for(int k=1;k<=m;k++)
                s[i][j][k]+=s[i+1][j][k],
                sum[i][j][k]+=sum[i+1][j][k];
    while(Q--)
    {
        int x1=sc(),y1=sc(),x2=sc(),y2=sc(),H=sc();
        int l=1,r=mx,ans=0;
        while(l<=r)
        {
            int mid=l+r>>1;
            int res=SUM(mid,x1,y1,x2,y2);
            if(res>=H)ans=mid,l=mid+1;else r=mid-1;
        }
        if(ans)
        {
            int res=SUM(ans,x1,y1,x2,y2);
            int tot=s[ans][x2][y2]-s[ans][x1-1][y2]-s[ans][x2][y1-1]+s[ans][x1-1][y1-1];
            while(res-ans>=H)res-=ans,tot--;
            printf("%d\n",tot);
        }else puts("Poor QLW");
    }
}
int main()
{
    n=sc(),m=sc(),Q=sc();
    if(n==1)solve1();else solve2();
    return 0;
}