Bzoj 1926: [Sdoi2010]粟粟的書架(二分答案+亂搞+主席樹)

1926: [Sdoi2010]粟粟的書架
Time Limit: 30 Sec Memory Limit: 552 MB
Description
幸福幼兒園 B29 班的粟粟是一個聰明機靈、乖巧可愛的小朋友,她的愛好是畫畫和讀書,尤其喜歡 Thomas H. Co
rmen 的文章。粟粟家中有一個 R行C 列的巨型書架,書架的每一個位置都擺有一本書,上數第i 行、左數第j 列
擺放的書有Pi,j頁厚。粟粟每天除了讀書之外,還有一件必不可少的工作就是摘蘋果,她每天必須摘取一個指定的
蘋果。粟粟家果樹上的蘋果有的高、有的低,但無論如何憑粟粟自己的個頭都難以摘到。不過她發現, 如果在腳
下放上幾本書,就可以夠着蘋果;她同時注意到,對於第 i 天指定的那個蘋果,只要她腳下放置書的總頁數之和
不低於Hi,就一定能夠摘到。由於書架內的書過多,父母擔心粟粟一天內就把所有書看完而耽誤了上幼兒園,於是
每天只允許粟粟在一個特定區域內拿書。這個區域是一個矩形,第 i 天給定區域的左上角是上數第 x1i行的左數
第 y1i本書,右下角是上數第 x2i行的左數第y2i本書。換句話說,粟粟在這一天,只能在這﹙x2i-x1i+1﹚×﹙
y2i-y1i+1﹚本書中挑選若干本墊在腳下,摘取蘋果。粟粟每次取書時都能及時放回原位,並且她的書架不會再
撤下書目或換上新書,摘蘋果的任務會一直持續 M天。給出每本書籍的頁數和每天的區域限制及採摘要求,請你告
訴粟粟,她每天至少拿取多少本書,就可以摘到當天指定的蘋果。
Input
第一行是三個正整數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。
Output
有M行,第i 行回答粟粟在第 i 天時爲摘到蘋果至少需要 拿取多少本書。如果即使取走所有書都無法摘到蘋果,
則在該行輸出“Poor QLW” (不含引號)。
Sample Input
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
Sample Output
6
15
2
Poor QLW
9
1
3
HINT
對於 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
Source
第一輪Day2

/*
SD的複合題2333.
前50%沒想出來,對於某些小的東西就應該多考慮orz.
K<=1000 顯然要亂搞啊...
b[i][j][k]表示(1,1)到(i,j)大於等於K的數的個數.
sum[i][j][k]表示表示(1,1)到(i,j)大於等於K的數的和.
然後二分答案.
注意ans處不一定要全選,樣例還是挺良心的。。。 
複雜度O(R*C*1000)+O(M*Log1000).
後50%的數據想出來了,很好想 
用主席樹維護區間前K大之和
然後二分答案檢驗.
複雜度O(M*Log2N).
*/
#include<algorithm>
#include<cstdio>
#define MAXN 500001
using namespace std;
int n,m,q,tot1,root[MAXN],b[201][201][1010],sum[201][201][1010],ans,Sum[MAXN],s[201][201],c[201][201];
int H,tot,total,l,r;
struct data{int lc,rc,size,sum;}tree[MAXN*20];
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
    return x*f;
}
void add(int &now,int last,int l,int r,int p)
{
    now=++tot1;tree[now].lc=tree[last].lc;
    tree[now].rc=tree[last].rc;
    tree[now].size=tree[last].size+1;
    tree[now].sum=tree[last].sum+p;
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(p<=mid) add(tree[now].lc,tree[last].lc,l,mid,p);
    else add(tree[now].rc,tree[last].rc,mid+1,r,p);
    return ;
}
int query(int L,int R,int l,int r,int k)
{
    if(l==r) return (tree[R].sum-tree[L].sum)/(tree[R].size-tree[L].size)*k;
    int sum1=tree[tree[R].lc].size-tree[tree[L].lc].size;
    int mid=(l+r)>>1;
    if(sum1>=k) return query(tree[L].lc,tree[R].lc,l,mid,k);
    else return query(tree[L].rc,tree[R].rc,mid+1,r,k-sum1)+
    tree[tree[R].lc].sum-tree[tree[L].lc].sum;
}
void slovequery(int x,int y)
{
    ans=0;
    l=1,r=y-x+1;int mid,k,L,X;
    L=y-x+1;X=Sum[y]-Sum[x-1];
    while(l<=r)
    {
        mid=(l+r)>>1;
        k=L-mid+1;
        if(X-query(root[x-1],root[y],1,1000,k)>=H) ans=mid-1,r=mid-1;
        else l=mid+1;
    }
    if(!ans) ans=y-x+1;
    printf("%d\n",ans);
}
void slove1()
{
    int x,x1,y1,x2,y2;
    for(int i=1;i<=m;i++)
    {
        x=read();
        Sum[i]=Sum[i-1]+x;
        add(root[i],root[i-1],1,1000,x);
    }
    while(q--)
    {
        x1=read(),y1=read(),x2=read(),y2=read();H=read();
        if(Sum[y2]-Sum[y1-1]<H) printf("Poor QLW\n");
        else if(Sum[y2]-Sum[y1-1]==H) printf("%d\n",y2-y1+1);
        else slovequery(y1,y2);
    }
}
int slovequery2(int x1,int y1,int x2,int y2)
{
    l=1,r=1000;int mid;ans=0;
    while(l<=r)
    {
        mid=(l+r)>>1;
        tot=sum[x2][y2][mid]-sum[x1-1][y2][mid]-sum[x2][y1-1][mid]+sum[x1-1][y1-1][mid];
        if(tot>=H) ans=mid,l=mid+1;
        else r=mid-1;
    }
    total=sum[x2][y2][ans]-sum[x1-1][y2][ans]-sum[x2][y1-1][ans]+sum[x1-1][y1-1][ans]-H;
    tot=b[x2][y2][ans]-b[x1-1][y2][ans]-b[x2][y1-1][ans]+b[x1-1][y1-1][ans];
    ans=tot-total/ans;
    return ans;
}
void slove2()
{
    int x1,y1,x2,y2;
    for(int i=1;i<=n;i++)
      for(int j=1;j<=m;j++)
        c[i][j]=read();
    for(int i=1;i<=n;i++)
      for(int j=1;j<=m;j++)
        s[i][j]=s[i][j-1]+s[i-1][j]+c[i][j]-s[i-1][j-1];
    for(int i=1;i<=n;i++)
      for(int j=1;j<=m;j++)
        for(int k=1;k<=1000;k++)
        {
            if(c[i][j]>=k) b[i][j][k]++,sum[i][j][k]=c[i][j];
            b[i][j][k]+=b[i][j-1][k]+b[i-1][j][k]-b[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];
        }
    while(q--)
    {
        x1=read(),y1=read(),x2=read(),y2=read();H=read();
        if(s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1]<H) printf("Poor QLW\n");
        else printf("%d\n",slovequery2(x1,y1,x2,y2));
    }
}
int main()
{
    freopen("susu.in","r",stdin);
    freopen("susu.out","w",stdout);
    n=read(),m=read(),q=read();
    if(n==1) slove1();
    else slove2();
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章