UPC-2020春季个人训练赛第四场(新生场)--E 黑匣子 (双堆,对顶堆)

参考博客
化身孤岛的鲸o
phython96
问题描述
Black Box是一种原始的数据库。它可以存储一个整数数组,还有一个特别的变量i。最开始的时候Black Box是空的,而i等于0。这个Black Box要处理一串命令。
命令只有两种:
ADD(x):把x元素放进Black Box;
GET:i加1,然后输出Black Box中第i小的数。
记住:第i小的数,就是Black Box里的数按从小到大的顺序排序后的第i个元素。
例如: 我们来演示一下一个有11个命令的命令串。(如下图所示)
序号 操作 i 数据库 输出
1 ADD(3) 0 3
2 GET 13 3
3 ADD(1) 11, 3
4 GET 21,3 3
5 ADD(-4) 2 -4,1, 3
6 ADD(2) 2-4,1,2,3
7 ADD(8) 2-4,1,2,3,8
8 ADD(-1000)2-1000,-4,1,2,3,8
9 GET 3 -1000,-4,1,2,3,81
10 GET 4 -1000,-4,1,2,3,8 2
11 ADD(2) 4-1000,-4,1,2,2,3,8

现在要求找出对于给定的命令串的最好的处理方法。ADD和GET命令分别最多有200000个。

现在用两个整数数组来表示命令串:
1、A(1),A(2),…A(M):一串将要被放进Black Box的元素。每个数都是绝对不超过2000000000的整数,M≤200000。例如上面的例子就是
A=(3,1,-4,2,8,-1000,2)。
2、u(1),u(2),…u(N):表示第u(j)个元素被放进了Black Box里后就出现了一个GET命令。例如上面的例子中的u=(1,2,6,6)。
输入数据不用判错。

输入
第一行,两个整数,M,N。
第二行,M个整数,表示A(1) …A(M)。
第三行,N个整数,表示u(1)…u(N)。
输出
输出Black Box根据命令串所得出的输出串,一个数字一行。
样例输入 Copy
7 4
3 1 -4 2 8 -1000 2
1 2 6 6
样例输出 Copy
3
3
1
2
提示
对于30%的数据,M≤10000;
对于50%的数据,M≤100000
对于100%的数据,M≤200000
题目分析
在这里插入图片描述化身孤岛的鲸o
这里我们用两个堆维护区间第K大的值,
上面的是小根堆
下面的是大根堆
如果大根堆中始终有k-1个元素,且小根堆中所有的元素都比大根堆中的大,那么小根堆的堆顶就是区间第k大的元素;
维护过程:
1.不停的往大根堆中加入数,使大根堆中的数的个数为k-1个,如果多了就把堆顶放到小根堆中
2.当小根堆的堆顶被访问过一次后k++,把小根堆的堆顶重新放回大根堆,这样使大根堆中的元素始终保持k-1个

**

ACcode

**

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <vector>
#include <iomanip>
#define  UpMing main
#define re register
#define Accept return 0;
#pragma GCC optimize(2)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf =0x3f3f3f3f;
const ll  inf2 =0x3f3f3f3f3f3f3f;
const int maxn=1e6+10;
const double eps=1e-7;
const ll mod = 20011021;
const int N =1e6+10;
const long double PI=3.1415926535897932384626433832795;
const long double e=2.71828182845904523536028747135266;
typedef pair<int,int>PII;
inline ll read() {
    
    
    ll  x=0;
    bool f=0;
    char ch=getchar();
    while (ch<'0'||'9'<ch)    f|=ch=='-', ch=getchar();
    while ('0'<=ch && ch<='9')
        x=x*10+ch-'0',ch=getchar();
    return f?-x:x;
}
priority_queue<ll,vector<ll>, greater<ll> >heap;
priority_queue<ll,vector<ll>, greater<ll> >mi;
priority_queue<ll >ma;
ll n,m,a[maxn],b[maxn],c[maxn],vis[maxn];
set<ll>s;
set<ll>::iterator it;
int UpMing() {
    
    
    m=read();
    n=read();
    ll cnt=1,th=0;
    for(int i=1 ; i<=m ; i++) a[i]=read();
    for(int i=1 ; i<=n; i++) b[i]=read(),vis[b[i]]++;
    for(int i=1 ; i<=m ; i++) {
    
    
        ma.push(a[i]);
        if(ma.size()>th) {
    
    
            mi.push(ma.top());
            ma.pop();
        }
        while(vis[i]) {
    
    
    printf("%lld\n",mi.top());
            ma.push(mi.top());
            mi.pop();
            th++;
            cnt++;
            vis[i]--;
        }
    }
  Accept;
}

猜你喜欢

转载自blog.csdn.net/wmy0536/article/details/105316921