洛谷-P1801 黑匣子_NOI导刊2010提高(06)

这个题就是让模拟一个数据库,有add和get两个操作,输入m,n;m代表一共添加m个树,n代表有n个get操作,给一共数组,b[i]表示,在添加进去前b[i]个数后输出第i小的。
我刚开始看到的时候想直接写一共vector,把元素按大小插入,然后直接输出就行了,倒是这么写也能过,但是可能慢一点吧
代码:

#include <stdio.h>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn = 200005;

int a[maxn],b[maxn];
vector<int> v;

int main()
{
	int m,n;
	scanf("%d%d",&m,&n);
	for(int i = 1;i <= m;i++){
		scanf("%d",&a[i]);
	}
	for(int i = 1;i <= n;i++){
		scanf("%d",&b[i]);
	}
	int k=1;
	int pos=0; 
	for(int i = 1;i <= m;i++){
		v.insert(lower_bound(v.begin(),v.end(),a[i]),a[i]);
		while(b[k] == i){
			printf("%d\n",v[pos]);
			pos++;
			k++;
		}
	}
	return 0;
}

后来才知道这题要用平衡树,不会啊。。。
看了很多大佬的题解,懂了对顶堆的写法,用两个优先队列维护,一个做小顶堆,一共做大顶堆,让大顶堆中留下i-1个元素,剩下的放在小顶堆中,堆顶就是当前要get的数值。
代码:

#include <stdio.h>
#include <algorithm>
#include <queue>
using namespace std;
const int maxn = 200005;

int a[maxn],b[maxn];

int main()
{
	int m,n;
	scanf("%d%d",&m,&n);
	for(int i = 1;i <= m;i++){
		scanf("%d",&a[i]);
	}
	priority_queue<int> q;
	priority_queue<int,vector<int>,greater<int> > q1;
	int l=1,r;
	for(int i = 1;i <= n;i++){
		scanf("%d",&r);
		for(int j = l;j <= r;j++){
			q.push(a[j]);
			if(q.size() == i){
				q1.push(q.top());
				q.pop();
			}
		}
		l = r+1;
		printf("%d\n",q1.top());
		q.push(q1.top());
		q1.pop();
	}
	return 0;
}


上面AC的是对顶堆,下面AC的是我原来的写法,的确差了很长时间。。。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值