【洛谷P1801】黑匣子_NOI导刊2010提高(06)(权值线段树裸题+模拟)

本文详细解析洛谷P1801 BlackBox问题,介绍了一种利用离散化和线段树解决大规模数据处理的高效算法。通过实际案例,展示了如何处理多达20万条的ADD和GET命令,实现对整数数组的有效管理和快速查询。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目地址:https://www.luogu.org/problem/P1801

题目


Black Box是一种原始的数据库。它可以储存一个整数数组,还有一个特别的变量i。最开始的时候Black Box是空的.而i等于0。这个Black Box要处理一串命令。

命令只有两种:

ADD(x):把x元素放进BlackBox;

GET:i加1,然后输出Blackhox中第i小的数。

记住:第i小的数,就是Black Box里的数的按从小到大的顺序排序后的第i个元素。例如:

我们来演示一下一个有11个命令的命令串。(如下图所示)

现在要求找出对于给定的命令串的最好的处理方法。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=(l,2,6,6)。输入数据不用判错。

输入格式

第一行,两个整数,M,N。

第二行,M个整数,表示A(l)……A(M)。

第三行,N个整数,表示u(l)…u(N)。

输出格式

输出Black Box根据命令串所得出的输出串,一个数字一行。

输入输出样例

说明/提示

对于30%的数据,M≤10000;

对于50%的数据,M≤100000:

对于100%的数据,M≤200000。

 

ac代码:


 

#include <bits/stdc++.h>
using namespace std;
const int maxn = 4*200005;
int a[maxn], b[maxn], cnt[maxn], u[maxn];
int q, m, n;;
void update(int id )
{
    cnt[id] = cnt[id << 1] + cnt[id << 1 | 1];
}
void build(int l, int r, int id, int no)
{
    if(l == r)
    {
        cnt[id]++;
        return ;
    }
    int mid = (l + r) >> 1;
    if(no <= mid) build(l, mid, id << 1, no);// 单点
    else build(mid + 1, r, id << 1 | 1, no);
    update(id);
}
int query(int l, int r, int id, int k)
{
    if(l == r)
        return l; //返回1-n的第k大的数是去重之后的第几大数
    int mid = (l + r) >> 1;
    if(k <= cnt[id << 1]) return  query(l, mid, id << 1, k);
    else return  query(mid + 1, r, id << 1 | 1, k - cnt[id << 1]);//!k - cnt[id << 1]
}
int main()
{
    //freopen("/Users/zhangkanqi/Desktop/11.txt","r",stdin);
    scanf("%d %d", &m, &n);
    for(int i = 1; i <= m; i++)
    {
        scanf("%d", &a[i]);
        b[i] = a[i];
    }
    for(int i = 1; i <= n; i++)
        scanf("%d", &u[i]);
    sort(a+1, a+1+m);
    int num = unique(a+1, a+1+m) - (a+1);
    for(int i = 1; i <= n; i++)
    {
        for(int j = u[i - 1] + 1; j <= u[i]; j++)
        {
            int no = lower_bound(a+1, a+1+num, b[j]) - a;
            build(1, num, 1, no);
        }
        int id = query(1, num, 1, ++q);
        printf("%d\n", a[id]);
    }
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值