Luogu: P1168 median (stl, binary heap and ST table, popularity +/improvement)

topic:

Analysis: I only thought of optimizing and modifying the original sorting algorithm to realize this problem.

Big guy stl:

#include<bits/stdc++.h>
using namespace std;
int m;
int main()
{
    
    
 cin>>m;
 vector<int> v;
 for(int i=0;i<m;i++) 
 {
    
    
  int c; cin>>c; 
  v.insert(lower_bound(v.begin(),v.end(),c),c);
  if(v.size()%2==1) cout<<v[v.size()/2]<<endl; 
 }
}

Of course upper is also possible.

But this question mainly wants to use the heap to achieve, try using the priority queue!

AC：

#include<bits/stdc++.h>
using namespace std;
int m;
int main()
{
    
    
 priority_queue<int,vector<int>,less<int> > q1;//存放数据小的一组，因为出队列是最大的出去。
 priority_queue<int,vector<int>,greater<int> > q2;//存放数据大的一组，因为出队列是最小的出去。 
 cin>>m;
 int a,b;
 cin>>a>>b;
 cout<<a<<endl;
 q1.push(min(b,a));
 q2.push(max(b,a));
 for(int i=2;i<m;i++)
 {
    
    
  cin>>a;
  if(a>=q2.top())
  {
    
    
   q2.push(a);
   int q2size=q2.size();
   int q1size=q1.size();
   if(q2size==q1size+2)
   {
    
    
    q1.push(q2.top());
    q2.pop();
   }
  }
  else{
    
    
   q1.push(a);
   int q2size=q2.size();
   int q1size=q1.size();
   if(q1size==q2size+2)
   {
    
    
    q2.push(q1.top());
    q1.pop();
   }
  }
  int q2size=q2.size();
  int q1size=q1.size();
  if(q1size!=q2size)
  {
    
    
   if(q1size==q2size+1) cout<<q1.top()<<endl;
   else cout<<q2.top()<<endl;
  }
 }  
}