zoukankan      html  css  js  c++  java
  • 栗栗的书架--洛谷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;
    }
    路漫漫兮
  • 相关阅读:
    .Net 并发写入文件的多种方式
    变量命名神器——CODELF
    Python打包发布
    python工具——NumPy
    python工具——Pandas
    没事早点睡
    python工具——pixellib
    Tensorflow在Windows下使用踩坑
    python工具——Tesseract
    python工具——wordcloud
  • 原文地址:https://www.cnblogs.com/lonely-wind-/p/11877532.html
Copyright © 2011-2022 走看看