思路:使用两个结构体优先队列。一个大根堆,存比Mid大的数字,方便及时取出大于Mid的第一个数(大根堆小顶);一个小根堆,存比Mid小的数字,方便及时取出小于Mid的第一个数(小根堆大顶);使用vis数组,记录那些已经弹出的数字,在优先队列那里判断一下然后及时弹出就可以了。使用了deque模拟栈的操作过程。
坑:实现细节较多,不过讨论完其中一种情况,其它情况基本上都是复制+粘贴+小更改就行。(实现细节请见代码)
#include<bits/stdc++.h>
#define int long long
using namespace std;
int N,i,x,vis[100009],cnt;
struct nod{
int bh,val;
}t,Mid,d;
struct cmp1{
bool operator()(nod x,nod y){
return x.val<y.val;
}
};
struct cmp2{
bool operator()(nod x,nod y){
return x.val>y.val;
}
};
string s;
deque<nod>q;
priority_queue<nod,vector<nod>,cmp1>q1;
priority_queue<nod,vector<nod>,cmp2>q2;
signed main(){
//freopen("158.in","r",stdin);freopen("158.out