栗栗的书架--洛谷P2468【SDOI2010】&BZOJ 1926二分+前缀和+主席树

时间限制3.00s
 
内存限制500.00MB
 

题目描述

幸福幼儿园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”(不含引号)。

输入输出样例

输入
5 5 7
14 15 9 26 53
58 9 7 9 32
38 46 26 43 38
32 7 9 50 28
8 41 9 7 17
1 2 5 3 139
3 1 5 5 399
3 3 4 5 91
4 1 4 1 33
1 3 5 4 185
3 3 4 3 23
3 1 3 3 108
输出 
6
15
2
Poor QLW
9
1
3
输入
1 10 7
14 15 9 26 53 58 9 7 9 32
1 2 1 9 170
1 2 1 9 171
1 5 1 7 115
1 1 1 10 228
1 4 1 4 45704571
1 1 1 1 1
1 7 1 8 16
输出 
6
7
3
10
Poor QLW
1
2

说明/提示

【数据规模和约定】

对于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。


。。。。如果直接做的话也不是不可以,可以直接分块,数列的分块的基础知识在我以前的博客里面有,可以通过友情链接进入。不过还有一种常规做法就是前50%的数据用二分加前缀和,后50%的数据用主席树维护就好了。

我们可以很容易看出,要使得数量最少,那么我们每次取应该取页数最大的那本书。

前50%的数据我们使用两个三维数组前缀和维护前k大的总和以及数量,最后二分答案即可。

二维前k大值之和的维护如下:

for (int i=1; i<=n; i++)
    for (int j = 1; j <= m; j++) {
        in(a[i][j]);
        if (vis[a[i][j]]) continue;
        vis[a[i][j]]=1;
        lis[++use]=a[i][j];
    }
sort(lis+1,lis+1+use,cmp);
for (int k=1; k<=use; k++) {
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++) {
            num[i][j][k] = num[i - 1][j][k] + num[i][j - 1][k] - num[i - 1][j - 1][k];
            sum[i][j][k] = sum[i - 1][j][k] + sum[i][j - 1][k] - sum[i - 1][j - 1][k];
            if (a[i][j] < lis[k]) continue;
            num[i][j][k]++;
            sum[i][j][k]+=a[i][j];
        }
}

 接下来就是无脑二分了,不过二分完之后由于最后一种页数的书可能会多出来,所以我们需要特判最后一种页数的情况:

int pu=sum[x2][y2][ans-1]-sum[x2][y1-1][ans-1]-sum[x1-1][y2][ans-1]+sum[x1-1][y1-1][ans-1];//ans是二分完之后的最小的页数值 
int p=(h-pu)/lis[ans];//总值-前k-1大之和,剩下的就是需要用第k(ans)大的数来填补的空缺 
if ((h-pu)%lis[ans]) p++;
int cnt=num[x2][y2][ans-1]-num[x2][y1-1][ans-1]-num[x1-1][y2][ans-1]+num[x1-1][y1-1][ans-1];//前k-1大的数的数量 
printf ("%d
",p+cnt);

于是前50%的数据就很快解决了:

inline int ok(int x1,int y1,int x2,int y2,int x,int h)
{
    int tot=sum[x2][y2][x]-sum[x2][y1-1][x]-sum[x1-1][y2][x]+sum[x1-1][y1-1][x];
    if (tot<h) return 0;
    return 1;
}

inline void work1(int n,int m,int q)
{
    int use=0;
    for (int i=1; i<=n; i++)
        for (int j = 1; j <= m; j++) {
            in(a[i][j]);
            if (vis[a[i][j]]) continue;
            vis[a[i][j]]=1;
            lis[++use]=a[i][j];
        }
    sort(lis+1,lis+1+use,cmp);
    for (int k=1; k<=use; k++) {
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= m; j++) {
                num[i][j][k] = num[i - 1][j][k] + num[i][j - 1][k] - num[i - 1][j - 1][k];
                sum[i][j][k] = sum[i - 1][j][k] + sum[i][j - 1][k] - sum[i - 1][j - 1][k];
                if (a[i][j] < lis[k]) continue;
                num[i][j][k]++;
                sum[i][j][k]+=a[i][j];
            }
    }
    while(q--){
        int x1,x2,y1,y2,h;
        in(x1);in(y1);in(x2);in(y2); in(h);
        int l=1,r=use,mid;
        int ans=1005;
        while (l<=r){
            mid=(l+r)>>1;
            if (ok(x1,y1,x2,y2,mid,h)){
                ans=mid;
                r=mid-1;
            }
            else l=mid+1;
        }
        if (ans==1005) printf ("Poor QLW
");
        else {
            int pu=sum[x2][y2][ans-1]-sum[x2][y1-1][ans-1]-sum[x1-1][y2][ans-1]+sum[x1-1][y1-1][ans-1];
            int p=(h-pu)/lis[ans];
            if ((h-pu)%lis[ans]) p++;
            int cnt=num[x2][y2][ans-1]-num[x2][y1-1][ans-1]-num[x1-1][y2][ans-1]+num[x1-1][y1-1][ans-1];
            printf ("%d
",p+cnt);
        }
    }
}

后面的50%的数据我们想一下,如果用前50%的方法做的话,维护5e5 的序列然后在枚举1e3的页数。。。。果断T了

那么我们想一想,维护前k大的序列之和。。。主席树不是挺OK的吗,m*log(1000)的复杂度,完全OK。

套个主席树(区间第k大)的update的板子,然后改一改query部分就好了:

inline int update(int l,int r,int pos,int val,int ed)
{
    int rt=++tree_sum;
    int mid=(l+r)>>1;
    tree[rt].l=tree[ed].l;tree[rt].r=tree[ed].r;
    tree[rt].size=tree[ed].size+1;
    tree[rt].sum=tree[ed].sum+val;
    if (l==r) return rt;
    if (mid>=pos) tree[rt].l=update(l,mid,pos,val,tree[ed].l);
    else tree[rt].r=update(mid+1,r,pos,val,tree[ed].r);
    return rt;
}

inline int query(int l,int r,int L,int R,int val)
{
    int ans=0;
    while (l<r){
        int mid=(l+r)>>1;
        int x=tree[tree[R].r].sum-tree[tree[L].r].sum;
        if (x<val){
            ans+=tree[tree[R].r].size-tree[tree[L].r].size;
            val-=x;r=mid,L=tree[L].l;R=tree[R].l;
        }
        else l=mid+1,L=tree[L].r,R=tree[R].r;
    }
    ans+=(val+l-1)/l;
    return ans;
}

update部分就不多说了,前一篇博客刚好讲到了,遍历经过的所有路径都会有所修改。

至于query部分,这里写的是非递归方式,当然递归的方式可能比较好理解。我们的每个树的节点分为左右节点,由于是先取大的,所以我们先判断右子树的值是否大于需要的高度,如果否,说明需要更多的书,那么我们的ans值就需要+右子树的大小,然后往左子树递归,如果是的话我们直接往右子树递归就好了。最后所得的val的值就是剩下第k大的值所要填补的空缺。为了保证向上取整,我们可以先把val+l(第k大的值)-1再除以l就好了

以下是AC代码:

#include <bits/stdc++.h>
using namespace std;

const int mac=5e5+10;

int a[205][205],sum[205][205][1005],num[205][205][1005];
int vis[1005],lis[1005];
struct node
{
    int sum,size,l,r;
}tree[mac*11];

inline void in(int &x)
{
    int f=0;
    char ch=getchar();
    while (ch>'9' || ch<'0') ch=getchar();
    while (ch>='0' && ch<='9') f=(f<<1)+(f<<3)+ch-'0',ch=getchar();
    x=f;
}

inline int ok(int x1,int y1,int x2,int y2,int x,int h)
{
    int tot=sum[x2][y2][x]-sum[x2][y1-1][x]-sum[x1-1][y2][x]+sum[x1-1][y1-1][x];
    if (tot<h) return 0;
    return 1;
}

inline bool cmp(int x,int y)
{
    return x>y;
}

int tree_sum=0,ver[mac];

inline int build(int l,int r)
{
    int rt=++tree_sum;
    if (l==r) return rt;
    int mid=(l+r)>>1;
    tree[rt].l=build(l,mid);
    tree[rt].r=build(mid+1,r);
    return rt;
}

inline int update(int l,int r,int pos,int val,int ed)
{
    int rt=++tree_sum;
    int mid=(l+r)>>1;
    tree[rt].l=tree[ed].l;tree[rt].r=tree[ed].r;
    tree[rt].size=tree[ed].size+1;
    tree[rt].sum=tree[ed].sum+val;
    if (l==r) return rt;
    if (mid>=pos) tree[rt].l=update(l,mid,pos,val,tree[ed].l);
    else tree[rt].r=update(mid+1,r,pos,val,tree[ed].r);
    return rt;
}

inline int query(int l,int r,int L,int R,int val)
{
    int ans=0;
    while (l<r){
        int mid=(l+r)>>1;
        int x=tree[tree[R].r].sum-tree[tree[L].r].sum;
        if (x<val){
            ans+=tree[tree[R].r].size-tree[tree[L].r].size;
            val-=x;r=mid,L=tree[L].l;R=tree[R].l;
        }
        else l=mid+1,L=tree[L].r,R=tree[R].r;
    }
    ans+=(val+l-1)/l;
    return ans;
}

inline void work1(int n,int m,int q)
{
    int use=0;
    for (int i=1; i<=n; i++)
        for (int j = 1; j <= m; j++) {
            in(a[i][j]);
            if (vis[a[i][j]]) continue;
            vis[a[i][j]]=1;
            lis[++use]=a[i][j];
        }
    sort(lis+1,lis+1+use,cmp);
    for (int k=1; k<=use; k++) {
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= m; j++) {
                num[i][j][k] = num[i - 1][j][k] + num[i][j - 1][k] - num[i - 1][j - 1][k];
                sum[i][j][k] = sum[i - 1][j][k] + sum[i][j - 1][k] - sum[i - 1][j - 1][k];
                if (a[i][j] < lis[k]) continue;
                num[i][j][k]++;
                sum[i][j][k]+=a[i][j];
            }
    }
    while(q--){
        int x1,x2,y1,y2,h;
        in(x1);in(y1);in(x2);in(y2); in(h);
        int l=1,r=use,mid;
        int ans=1005;
        while (l<=r){
            mid=(l+r)>>1;
            if (ok(x1,y1,x2,y2,mid,h)){
                ans=mid;
                r=mid-1;
            }
            else l=mid+1;
        }
        if (ans==1005) printf ("Poor QLW
");
        else {
            int pu=sum[x2][y2][ans-1]-sum[x2][y1-1][ans-1]-sum[x1-1][y2][ans-1]+sum[x1-1][y1-1][ans-1];
            int p=(h-pu)/lis[ans];
            if ((h-pu)%lis[ans]) p++;
            int cnt=num[x2][y2][ans-1]-num[x2][y1-1][ans-1]-num[x1-1][y2][ans-1]+num[x1-1][y1-1][ans-1];
            printf ("%d
",p+cnt);
        }
    }
}

inline void work2(int n,int m,int q)
{
    ver[0]=build(1,1000);
    for (int i=1; i<=m; i++){
        int x;in(x);
        ver[i]=update(1,1000,x,x,ver[i-1]);
    }
    while (q--){
        int x1,x2,y1,y2,h;
        in(x1);in(y1);in(x2);in(y2);in(h);
        int ans=0;
        if (tree[ver[y2]].sum-tree[ver[y1-1]].sum<h) {printf ("Poor QLW
");continue;}
        ans=query(1,1000,ver[y1-1],ver[y2],h);
        printf("%d
",ans);
    }
}

int main()
{
    //freopen("in.txt","r",stdin);
    int n,m,q;
    in(n);in(m);in(q);
    if (n>1) work1(n,m,q);
    else work2(n,m,q);
    return 0;
}
路漫漫兮
原文地址:https://www.cnblogs.com/lonely-wind-/p/11877532.html