目录
显示
题目链接
https://www.luogu.org/problem/P3871
题解
我们建立两个堆,一个大根堆,一个小根堆,且小根堆内的所有元素都比大根堆的元素要大。
刚开始把所有元素都插入同一个堆中。
每次要求中位数的时候,我们从元素较多的堆中取出几个元素,放入元素较少的堆中,直到大根堆有至少一半的元素为止,此时大根堆堆顶即为中位数的值。
插入一个数的时候,我们将这个数与大根堆堆顶作比较,如果比堆顶小则插入大根堆,否则插入小根堆。这样我们就能确保小根堆内的所有元素都比大根堆的元素要大了。
#include <cstdio>
#include <queue>
using namespace std;
priority_queue<int> q1;
priority_queue<int,vector<int>,greater<int> > q2;
int siz1,siz2;
char s[5];
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
q1.push(x);
}
siz1=n;
int q;
scanf("%d",&q);
while(q--)
{
scanf("%s",s);
if(s[0]=='a')
{
int x;
scanf("%d",&x);
if(x<=q1.top())q1.push(x),siz1++;
else q2.push(x),siz2++;
}
else
{
while(siz1>siz2)
{
q2.push(q1.top());
q1.pop();
siz1--,siz2++;
}
while(siz1<siz2)
{
q1.push(q2.top());
q2.pop();
siz2--,siz1++;
}
printf("%d\n",q1.top());
}
}
return 0;
}
#include <cstdio>
#include <queue>
using namespace std;
priority_queue<int> q1;
priority_queue<int,vector<int>,greater<int> > q2;
int siz1,siz2;
char s[5];
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
q1.push(x);
}
siz1=n;
int q;
scanf("%d",&q);
while(q--)
{
scanf("%s",s);
if(s[0]=='a')
{
int x;
scanf("%d",&x);
if(x<=q1.top())q1.push(x),siz1++;
else q2.push(x),siz2++;
}
else
{
while(siz1>siz2)
{
q2.push(q1.top());
q1.pop();
siz1--,siz2++;
}
while(siz1<siz2)
{
q1.push(q2.top());
q2.pop();
siz2--,siz1++;
}
printf("%d\n",q1.top());
}
}
return 0;
}
#include <cstdio> #include <queue> using namespace std; priority_queue<int> q1; priority_queue<int,vector<int>,greater<int> > q2; int siz1,siz2; char s[5]; int main() { int n; scanf("%d",&n); for(int i=1;i<=n;i++) { int x; scanf("%d",&x); q1.push(x); } siz1=n; int q; scanf("%d",&q); while(q--) { scanf("%s",s); if(s[0]=='a') { int x; scanf("%d",&x); if(x<=q1.top())q1.push(x),siz1++; else q2.push(x),siz2++; } else { while(siz1>siz2) { q2.push(q1.top()); q1.pop(); siz1--,siz2++; } while(siz1<siz2) { q1.push(q2.top()); q2.pop(); siz2--,siz1++; } printf("%d\n",q1.top()); } } return 0; }