3784 对顶堆模板题

知识点应用

#include<cstdio>
#include<queue>
#include<functional>
using namespace std;

priority_queue<int> q1; //大根堆Q1：维护集合中较小值的部分的最大值。


priority_queue<int, vector<int>, greater<int> > q2; //小根堆Q2：维护集合中较大值的部分的最小值。

void insert(int x)
{
  if (!q2.size() || x > q2.top()) 
    q2.push(x);
  else 
    q1.push(x);
  if (q1.size() > q2.size() + 1) { //保证两个堆元素数量相差不超过一
    q2.push(q1.top()); 
    q1.pop(); 
  }
  if (q2.size() > q1.size() + 1) { 
    q1.push(q2.top()); 
    q2.pop(); 
  }
}

int main()
{
  int T, n, id, mid;
  scanf("%d", &T);

  while (T--)
  {
    while (q1.size()) 
      q1.pop();
    while (q2.size()) 
      q2.pop();
    scanf("%d%d", &id, &n);
    printf("%d %d\n", id, (n + 1) / 2);

    int cnt = 0;
    for (int i = 1; i <= n;i++)
    {
      int x; scanf("%d", &x);
      insert(x);
      if (i & 1)
      {
        printf("%d ", q1.size() > q2.size() ? q1.top() : q2.top());//奇数的时候多出来的就是中位数
        if (++cnt == 10 && i != n) { 
          puts(""); 
          cnt -= 10; 
        } //输出格式要注意 
      }
    }
    puts("");
  }
  return 0;
}