数据结构:从堆到“漏斗”

本文介绍了如何利用堆解决Black Box问题,这是一个涉及到处理整数数组和查找第i小元素的问题。通过对问题的分析,提出了使用两个堆(大顶堆和小顶堆)形成类似漏斗的结构,有效处理ADD和GET命令,避免了常规排序的高复杂度。文章通过实例展示了这种方法的思路,并提到该方法可用PQ模板实现。
摘要由CSDN通过智能技术生成

问题引入:黑匣子

Black Box是一种原始的数据库。它可以储存一个整数数组,还有一个特别的变量i。最开始的时候Black Box是空的.而i等于0。这个Black Box要处理一串命令。
命令只有两种:
ADD(x):把x元素放进BlackBox;
GET:i加1,然后输出Blackhox中第i小的数。
记住:第i小的数,就是Black Box里的数的按从小到大的顺序排序后的第i个元素。
现在要求找出对于给定的命令串的最好的处理方法。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)个元素被放进了Blaek Box里后就出现一个GET命令。例如上面的例子中u=(l,2,6,6)。输入数据不用判错。
输入输出格式
输入格式:
第一行,两个整数,M,N。
第二行,M个整数,表示A(l)……A(M)。
第三行,N个整数,表示u(l)…u(N)。
输出格式:
输出Black Box根据命令串所得出的输出串,一个数字一行。


这道题来源于2010年的NOI导刊。看到这题的感受如何?不就是排序吗?有道理,我们来排排看!
假设我们用O(n log n)的算法……最多应付……3000数据……
那么,这种慢慢插入的方法,是不是能令人想到堆呢?没错,其实就是一个不断进行堆操作的过程。
不过,想到这一步,还是眼泪掉下来……
堆的性质不能保证数列有序啊!
但是仔细想想,这题参与操作的无非就是恰在某条分割线上位置的数据……每一次Get就是移动这条分割线,Add就是往分割线处插数据……

突然有一个漏斗的形状浮现在脑海中!

没错……就是一个漏斗!两个堆,像漏斗一样地摆放。上面是大顶堆,下面是小顶堆!大功告成。
【事实上这题可以用PQ模版】

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
#define read(x) scanf("%d",&x)
#define write(x) printf("%d\n",x)
int a[200005],u[200005],na,nu;
vector <int>vb,vs; 
int add(int x)
{
    int tmp;
    if(vb.empty()||x>vb[0])
        vs.push_back(x),
        push_heap(vs.begin(),vs.end(),greater<int>());
    else
        tmp=vb[0],
        pop_heap(vb.begin(),vb.end()),
        vb.pop_back(),
        vb.push_back(x),
        push_heap(vb.begin(),vb.end()),
        vs.push_back(tmp),
        push_heap(vs.begin(),vs.end(),greater<int>());
    return 1;
}
void get()
{
    int tmp=vs[0];
    pop_heap(vs.begin(),vs.end(),greater<int>());
    vs.pop_back();
    vb.push_back(tmp);
    push_heap(vb.begin(),vb.end());
    write(tmp);
    return;
}
int main()
{
    read(na);read(nu);
    int i,j,t;
    for(i=1;i<=na;i++) read(a[i]);
    for(i=1;i<=nu;i++) read(u[i]);
    i=0;j=1;
    while(++i<=na&&add(a[i]))
        while(i==u[j]) get(), j++;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值