洛谷P2468 [SDOI2010]粟粟的书架【主席树】

时空限制 3000ms / 512MB

题目描述

幸福幼儿园B29班的粟粟是一个聪明机灵、乖巧可爱的小朋友，她的爱好是画画和读书，尤其喜欢Thomas H. Cormen的文章。粟粟家中有一个R行C列的巨型书架，书架的每一个位置都摆有一本书，上数第i行、左数第j列摆放的书有Pi,j页厚。

粟粟每天除了读书之外，还有一件必不可少的工作就是摘苹果，她每天必须摘取一个指定的苹果。粟粟家果树上的苹果有的高、有的低，但无论如何凭粟粟自己的个头都难以摘到。不过她发现，如果在脚下放上几本书，就可以够着苹果；她同时注意到，对于第i天指定的那个苹果，只要她脚下放置书的总页数之和不低于Hi，就一定能够摘到。

由于书架内的书过多，父母担心粟粟一天内就把所有书看完而耽误了上幼儿园，于是每天只允许粟粟在一个特定区域内拿书。这个区域是一个矩形，第i天给定区域的左上角是上数第x1i行的左数第y1i本书，右下角是上数第x2i行的左数第y2i本书。换句话说，粟粟在这一天，只能在这﹙x2i－x1i＋1﹚×﹙y2i－y1i＋1﹚本书中挑选若干本垫在脚下，摘取苹果。

粟粟每次取书时都能及时放回原位，并且她的书架不会再撤下书目或换上新书，摘苹果的任务会一直持续M天。给出每本书籍的页数和每天的区域限制及采摘要求，请你告诉粟粟，她每天至少拿取多少本书，就可以摘到当天指定的苹果。

输入格式：

输入文件susu.in第一行是三个正整数R, C, M。
接下来是一个R行C列的矩阵，从上到下、从左向右依次给出了每本书的页数Pi,j。
接下来M行，第i行给出正整数x1i, y1i, x2i, y2i, Hi，表示第i天的指定区域是﹙x1i, y1i﹚与﹙x2i, y2i﹚间的矩形，总页数之和要求不低于Hi。

保证1≤x1i≤x2i≤R，1≤y1i≤y2i≤C。

输出格式：

输出文件susu.out有M行，第i行回答粟粟在第i天时为摘到苹果至少需要拿取多少本书。如果即使取走所有书都无法摘到苹果，则在该行输出“Poor QLW”（不含引号）。

说明

对于10%的数据，满足R, C≤10；
对于20%的数据，满足R, C≤40；
对于50%的数据，满足R, C≤200，M≤200,000；
另有50%的数据，满足R＝1，C≤500,000，M≤20,000；
对于100%的数据，满足1≤Pi,j≤1,000，1≤Hi≤2,000,000,000

---

题目分析

一开始脑袋被门夹了想了好久怎么用主席树维护二维数组，然后发现可以按数据分开做。。。
R=1的情况建个主席树，二分k判断区间内最厚的k本书和是否大于h即可

另外50%的情况感觉时限卡的好死，好像也没什么特别正确的思路
val[i][j][k]记录以i,j为右下角的矩阵内大于等于k的数的和，cnt[i][j][k]记录个数，也是二分
BZOJ上被卡常了，上网搜了一下好像也没什么其他好思路，就当我脸黑吧

#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long lt;

lt read()
{
    lt f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return x*f;
}

const int lim=1000;
const int maxn=500010;
int n,m,Q;
int rt[maxn],ch[maxn<<5][2],size[maxn<<5],sz;
lt sum[maxn<<5];

int update(int pre,int ll,int rr,int x)
{
    int tt=++sz,mid=ll+rr>>1; 
    size[tt]=size[pre]+1; sum[tt]=sum[pre]+x;
    ch[tt][0]=ch[pre][0]; ch[tt][1]=ch[pre][1];
    if(ll<rr)
    {
        if(x<=mid) ch[tt][0]=update(ch[pre][0],ll,mid,x);
        else ch[tt][1]=update(ch[pre][1],mid+1,rr,x);
    }
    return tt;
}

int qsum(int u,int v,int ll,int rr,int k)
{
    if(ll==rr) return ll*k;
    int mid=ll+rr>>1,ss=size[ch[v][1]]-size[ch[u][1]];
    if(k<=ss) return qsum(ch[u][1],ch[v][1],mid+1,rr,k);
    else return sum[ch[v][1]]-sum[ch[u][1]]+qsum(ch[u][0],ch[v][0],ll,mid,k-ss);
}

int query1(int ll,int rr,int H)
{
    if(sum[rt[rr]]-sum[rt[ll-1]]<H) return -1;
    int L=0,R=m+1,res=0;
    while(L<R)
    {
        int mid=L+R>>1;
        if(qsum(rt[ll-1],rt[rr],1,lim,mid)>=H) R=mid,res=mid;
        else L=mid+1;
    }
    return res;
}

void solve1()
{
    for(int i=1;i<=m;++i)
    rt[i]=update(rt[i-1],1,lim,read());
    
    while(Q--)
    {
        int a=read(),b=read(),c=read(),d=read(),H=read();
        int ans=query1(b,d,H);
        if(ans==-1) printf("Poor QLW\n");
        else printf("%d\n",ans);
    }
}

const int maxN=205;
int cnt[maxN][maxN][1010];
lt val[maxN][maxN][1010];
lt a[maxN][maxN];

int getV(int a,int b,int c,int d,int k){
    return val[c][d][k]-val[a-1][d][k]-val[c][b-1][k]+val[a-1][b-1][k];
}

int getC(int a,int b,int c,int d,int k){
    return cnt[c][d][k]-cnt[a-1][d][k]-cnt[c][b-1][k]+cnt[a-1][b-1][k];
}

int query2(int a,int b,int c,int d,int H)
{
    if(getV(a,b,c,d,0)<H) return -1;
    int L=0,R=lim+1,res=-1;
    while(L<R)
    {
        int mid=L+R>>1;
        if(getV(a,b,c,d,mid)>=H) L=mid+1,res=mid;
        else R=mid;
    }
    
    if(res==-1) return -1;
    else return getC(a,b,c,d,res)-(getV(a,b,c,d,res)-H)/res;
}

void solve2()
{
    for(int i=1;i<=n;++i)
    for(int j=1;j<=m;++j)
    a[i][j]=read();
    
    for(int k=0;k<=lim;++k)
    for(int i=1;i<=n;++i)
    for(int j=1;j<=m;++j)
    {
        val[i][j][k]=val[i-1][j][k]+val[i][j-1][k]-val[i-1][j-1][k]+(a[i][j]>=k?a[i][j]:0);
        cnt[i][j][k]=cnt[i-1][j][k]+cnt[i][j-1][k]-cnt[i-1][j-1][k]+(a[i][j]>=k?1:0);
    }
    
    while(Q--)
    {
        int a=read(),b=read(),c=read(),d=read(),H=read();
        int ans=query2(a,b,c,d,H);
        if(ans==-1) printf("Poor QLW\n");
        else printf("%d\n",ans);
    }
}

int main()
{
    n=read();m=read();Q=read();
    if(n==1) solve1();
    else solve2();
    return 0;
}