这个题就是让模拟一个数据库,有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的是我原来的写法,的确差了很长时间。。。