堆--P1168 中位数

 

记录一个变量$mid$，我们知道中位数是大小处于中间位置的数，所以建立两个堆，一个大根堆，一个小根堆，大根堆存≤ $mid$的数，小根堆存>$mid$的数。所以我们每次向堆中加入元素时，就通过比较和$mid$的大小关系，选择加入大根堆或者小根堆，但我们在输出答案前需要对$mid$进行调整。如果大根堆和小根堆内的元素个数相同，就无需处理，此时$mid$已然是当前的中位数。如果两个堆中元素个数不同，就需要取个数多的一个堆的堆顶，与$mid$取平均值

 #include <cstdio>
 #include <iostream>
 #include <queue>
 #include <algorithm>
 using namespace std;
 int n;
 int a[];
 int mid;
 priority_queue <int,vector<int>,less<int> >q1;
 priority_queue <int,vector<int>,greater<int> >q2;
 int main()
 {
     scanf ("%d",&n);
     scanf ("%d",&a[]);
     mid=a[];
     printf ("%d\n",mid);
     for (int i = ;i <= n;i++)
     {
         scanf ("%d",&a[i]);
         if (a[i]>mid) q2.push(a[i]);
         else q1.push(a[i]);
         if (i%==)
         {
             while (q1.size()!=q2.size())
             {
                 if(q1.size()>q2.size()){
                     q2.push(mid);
                     mid=q1.top();
                     q1.pop();
                 }
                 else{
                     q1.push(mid);
                     mid=q2.top();
                     q2.pop();
                 }
             }
             cout<<mid<<endl;
         }
             }
     return ;
 }