神奇的对顶堆
动态维护第K小
O(logn)单次
维护第k小
如图元素上大下小,保证大根堆中有k-1个元素
若大根堆元素个数=k,将大根堆堆顶弹入小根堆
每次取小根堆的堆顶即为答案。
例题:
洛谷P1801黑匣子
题目链接
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll a[200005];
int m,n,t,s=1,k;
priority_queue<ll> q1;
priority_queue<ll,vector<ll>,greater<ll> >q2;
int main()
{
cin>>m>>n;
for(int i=1;i<=m;i++)cin>>a[i];
while(n--)
{
cin>>t;
k++;
for(;s<=t;s++)
{
q1.push(a[s]);
if(q1.size()==k)
{
q2.push(q1.top());
q1.pop();
}
}
cout<<q2.top()<<endl;
q1.push(q2.top());
q2.pop();
s=t+1;
}
return 0;
}
洛谷P1168中位数
题目链接
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll a[100005];
int m,n,t,s=1,k;
priority_queue<ll> q1;
priority_queue<ll,vector<ll>,greater<ll> >q2;
int main()
{
cin>>m;
for(int i=1;i<=m;i++)
{
cin>>a[i];
q1.push(a[i]);
if(q1.size()>=i/2+1)
{
q2.push(q1.top());
q1.pop();
}
if(i&1)
{
cout<<q2.top()<<endl;
q1.push(q2.top());
q2.pop();
}
}
return 0;
}