洛谷 P2468 粟粟的書架 二分(主席樹+前綴和)

傳送~:https://www.luogu.org/problem/P2468

看了一下數據也發現是兩道題,後邊當他是一個序列(n==1)的時候直接主席樹二分區間前k大和就行了

但是有一個細節我覺得就是直接不去重的離散化會避免很多問題~所以直接按照最笨的方法排兩次序

然後按照順序添加主席樹查找就行。

但是那50%的數據n!=1有點不知道怎麼做了,一開始想到主席樹肯定不行,然後二位前綴和也搞不出來,然後看了題解後才知道,只需要開一個三維數組來記錄val[i][j][k]表示從1,1~i,j的這個區間有多少大於k的數就行了,主要是k他不大,可以直接全部處理出來,然後再開一個三維num[i][j][k]來記錄個數,然後可以通過二維前綴和的方式相減得到某一個矩形情況,二分即可得到答案~

代碼如下:

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>

#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define PI 3.14159265358979323846
#define min4(a, b, c, d) min(min(a,b),min(c,d))
#define min3(x, y, z) min(min(x,y),min(y,z))
#define pii make_pair
#define pr pair<int,int>
const int dir[4][2] = {0, -1, -1, 0, 0, 1, 1, 0};
typedef long long ll;
const ll inFF = 9223372036854775807;
typedef unsigned long long ull;
using namespace std;
const int maxn=5e5+4;
int val[205][205][1005];
int num[205][205][1005];
int b[205][205];
struct nod
{
    ll x;
    int pos,id;
}a[maxn];
bool cmp(nod s,nod e) {return s.x<e.x;}
bool cmp1(nod s,nod e) {return s.id<e.id;}
struct node
{
    int l,r,cnt;
    ll sum;
}tr[maxn*100];
int root[maxn];
int t;
int n,m,q;
int get_sum(int x1,int y1,int x2,int y2,int x)
{
    return val[x2][y2][x]-val[x1-1][y2][x]-val[x2][y1-1][x]+val[x1-1][y1-1][x];
}
int get_num(int x1,int y1,int x2,int y2,int x)
{
    return num[x2][y2][x]-num[x1-1][y2][x]-num[x2][y1-1][x]+num[x1-1][y1-1][x];
}
void work2()
{
    int ma=0;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            scanf("%d",&b[i][j]),ma=max(ma,b[i][j]);
    for(int k=0;k<=ma;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]+(b[i][j]>=k?b[i][j]:0);
                num[i][j][k]=num[i-1][j][k]+num[i][j-1][k]-num[i-1][j-1][k]+(b[i][j]>=k?1:0);
            }
    int x1,y1,x2,y2,h;
    while(q--)
    {
        scanf("%d %d %d %d %d",&x1,&y1,&x2,&y2,&h);
        if(get_sum(x1,y1,x2,y2,0)<h) {puts("Poor QLW");continue;}
        int l=0,r=ma,ans=-1;
        while(l<=r)
        {
            int mid=half;
            if(get_sum(x1,y1,x2,y2,mid)>=h)
            {
                ans=mid;
                l=mid+1;
            }
            else r=mid-1;
        }
        if(ans==-1) {puts("Poor QLW");continue;}
        else printf("%d\n",get_num(x1,y1,x2,y2,ans)-(get_sum(x1,y1,x2,y2,ans)-h)/ans);
        //這裏(get_sum(x1,y1,x2,y2,ans)-h)/ans是可能有很多大小=ans的數,這個時候可能多算了哈
        //所以說算出來sum減掉h就是多出來的,然後看是不是多了幾個大小爲ans的,把他們刪掉
    }
}
void insert(int &now,int pre,int l,int r,int k,ll x)
{
    now=++t;
    tr[now]=tr[pre];
    tr[now].cnt++;
    tr[now].sum+=x;
    if(l==r) return;
    int mid=half;
    if(k<=mid) insert(tr[now].l,tr[pre].l,l,mid,k,x);
    else insert(tr[now].r,tr[pre].r,mid+1,r,k,x);
}
ll query(int st,int ed,int l,int r,int k)//自己寫的主席樹好醜0A0
{
    if(k==0) return 0;
    if(l==r) return tr[ed].sum-tr[st].sum;
    int x=tr[tr[ed].r].cnt-tr[tr[st].r].cnt;
    ll ans=0;
    int mid=half;
    if(x<=k)
    {
        ans+=tr[tr[ed].r].sum-tr[tr[st].r].sum;
        ans+=query(tr[st].l,tr[ed].l,l,mid,k-x);
    }
    else ans=query(tr[st].r,tr[ed].r,mid+1,r,k);
    return ans;
}
void work1()
{
    n=m;
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i].x),a[i].id=i;
    sort(a+1,a+1+n,cmp);
    for(int i=1;i<=n;i++) a[i].pos=i;
    sort(a+1,a+1+n,cmp1);
    for(int i=1;i<=n;i++)
        insert(root[i],root[i-1],1,n,a[i].pos,a[i].x);
    int x1,y1,x2,y2;
    ll h;
    while(q--)
    {
        scanf("%d %d %d %d %lld",&x1,&y1,&x2,&y2,&h);
        if(query(root[y1-1],root[y2],1,n,y2-y1+1)<h) {puts("Poor QLW");continue;}
        int l=0,r=y2-y1+1,ans=-1;
        while(l<=r)
        {
            int mid=half;
            if(query(root[y1-1],root[y2],1,n,mid)>=h)
            {
                  ans=mid;
                  r=mid-1;
            }
            else
                l=mid+1;
        }
        printf("%d\n",ans);
    }
}
int main()
{
    cin>>n>>m>>q;
    if(n==1) work1();
    else work2();
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章