luogu P2343 宝石管理系统 |分块+堆

题目描述

GY君购买了一批宝石放进了仓库。有一天GY君心血来潮，想要清点他的宝石，于是把m个宝石都取出来放进了宝石管理系统。每个宝石i都有一个珍贵值vi，他希望你能编写程序查找到从大到小第n珍贵的宝石。但是现在问题来了，他非常不小心的留了一些宝石在仓库里面，有可能要往现有的系统中添加宝石。这些宝石的个数比较少。他表示非常抱歉，但是还是希望你的系统能起作用。

输入格式

第一行一个整数m，q，表示已经取出来的宝石个数以及接下来的查询或插入操作个数。

第二行m个整数，表示这m个宝石的珍贵值。

以下q行，每行两个整数c，n,

若c=1（即询问），则输出当前第n珍贵的宝石，

若c=2（即插入），则往系统中插入珍贵值为n的宝石。

输出格式

对于每个c=1（询问）,输出当前第n珍贵的宝石的珍贵值vi。

每个块里建一个堆

还可以用树状数组优化，但是没有写

---

#include<cmath>
#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define int long long
const int N=4e5+10;
int A[N],B[N],belong[N],len,num;
struct node{
    int c,x;
}e[N];
priority_queue<int>q[N];
int n,m;
inline void add(int x){// log(n)
    x=lower_bound(B+1,B+1+len,x)-B;
    int op=belong[x];
    q[op].push(x);
}
inline void ask(int x){// sqrt(n)+sqrt(n)*log(sqrt(n))
    int op;
    for(int i=num;i>=0;i--)
    if(x>q[i].size())x-=q[i].size();
    else { op=i; break; }
    vector<int>p;
    int ans;
    while(x){
        int u=q[op].top();q[op].pop();
        p.push_back(u);
        ans=u;
        x--;
    }
    printf("%lld\n",B[ans]);
    for(int i=0;i<p.size();i++)
    q[op].push(p[i]);
}
signed main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        scanf("%lld",&A[i]);
        B[i]=A[i];
    }
    int sum=n;
    for(int i=1;i<=m;i++){
        scanf("%lld%lld",&e[i].c,&e[i].x);
        if(e[i].c==2)B[++sum]=e[i].x;
    }
    sort(B+1,B+1+sum);
    len=unique(B+1,B+1+sum)-B-1;
    int size=pow(len,1.0/3.0);
    num=ceil((double)n/size);

    for(int i=1;i<=num;i++)
    for(int j=(i-1)*size+1;j<=size*i;j++)
    belong[j]=i;

    for(int i=1;i<=n;i++)add(A[i]);

    for(int i=1;i<=m;i++){
        int c=e[i].c,x=e[i].x;
        if(c==1)
            ask(x);
        else
            add(x);

    }
}