MT2016——数据流的中位数

数据流的中位数

大顶堆和小顶堆解决问题
大顶堆 ：每个结点的值都大于或等于其左右孩子结点的值。
小顶堆 ：每个结点的值都小于或等于其左右孩子结点的值。

#include<bits/stdc++.h> 
const int N=100000;
using namespace std;
priority_queue<int,vector<int>,less<int>>q1;//大顶堆，存储小一半的数
priority_queue<int,vector<int>,greater<int>>q2;//小顶堆，存储大一半的数
//大顶堆：每个结点的值都大于其左右孩子结点的值
void insert(int num)
{
     if(q1.size()>q2.size())//放入之前栈中元素为奇数个
     {
      q1.push(num);//入栈
      q2.push(q1.top());//将q1的栈顶元素插入q2中
      q1.pop();//q1栈顶元素出栈
     }
     else//放入之前栈中元素为偶数个
     {
      q1.push(num);
      if(q2.size()!=0 && q1.top()>q2.top())
      {
       q2.push(q1.top());q1.pop();
       q1.push(q2.top());q2.pop();
      }
     }
}
int main( )
{
    int n,num;
    cin>>n;
    for(int i=0;i<n;i++)
    {
     char ch;
     cin>>ch;
     if(ch=='+')//遇到数字
     {
         cin>>num;
         insert(num);
     }
     else{
         if(q1.size()>q2.size())
         {
             cout<<q1.top()<<endl;
         }
         else
         {
             cout<<(float)(q1.top()+q2.top())/2<<endl;
         }
     }
    }
    return 0;
}