牛客 Running Median (对顶堆)

For this problem, you will write a program that reads in a sequence of 32-bit signed integers. After each odd-indexed value is read, output the median (middle value) of the elements received so far.
输入描述:
The first line of input contains a single integer P(1 \leq P \leq 1000)P(1≤P≤1000), which is the number of data sets that follow. The first line of each data set contains the data set number, followed by a space, followed by an odd decimal integer M (1 \leq M \leq 9999)M(1≤M≤9999), giving the total number of signed integers to be processed. The remaining line(s) in the dataset consists of the values, 10 per line, separated by a single space. The last line in the dataset may contain less than 10 values.
输出描述:
For each data set the first line of output contains the data set number, a single space and the number of medians output (which should be one-half the number of input values plus one). The output medians will be on the following lines, 10 per line separated by a single space. The last line may have less than 10 elements, but at least 1 element. There should be no blank lines in the output.

输入

3
1 9
1 2 3 4 5 6 7 8 9
2 9
9 8 7 6 5 4 3 2 1
3 23
23 41 13 22 -3 24 -31 -11 -8 -7
3 5 103 211 -311 -45 -67 -73 -81 -99
-33 24 56

输出
1 5
1 2 3 4 5
2 5
9 8 7 6 5
3 12
23 23 22 22 13 3 5 5 3 -3
-7 -3

题目大意: 这道题目是求中位数,但是它和其他求中位数的不一样,我们平常求中位数是这样,如果是奇数,那就是最中间的那个数,如果是偶数,那就是中间的两位数和的平均值,但是这到题目不一样,他每次加入一个数的时候都要求当前数列的中位数,而且他的中位数是中间那个值并且是奇数。

题目分析

  • 这里我们需要用两个堆去存储数据,一个小顶堆(最大值在堆底),一个大顶堆(最大值在堆顶),这样我们称之为对顶堆,有了两个堆去存储数据,我们就能很快的求出中位数
  • 但是,我们可能会遇到一些情况,就是一个堆中的数据大大多于另外一个堆中的数据,这样我们就很难求出中位数,因为中位数不在堆顶,而我们要求中位数一定是要在两个堆的堆顶之一,所以这时候就需要数据转移,我们保证两个堆的数据数量相差不能超过2个,而且我们把中位数放在小顶堆的堆顶(读者可以自行选择),这样我们只需要每次取小顶堆的堆顶放在动态数组中,最后只需要遍历一次vector数组就可以把全部的中位数输出
  • 输出格式需要注意一下,每十个需要换行一下,而且最后还要判断是否有数据,因为是多组输入,所以这一点需要特别注意,简单来说就是每一行最后都要换行。
#include <bits/stdc++.h>
using namespace std;

vector<int> p;
// 保持小根堆中最小元素要大于大根堆中的全部元素
priority_queue<int, vector<int>, greater<int> > q1; //小根堆
priority_queue<int, vector<int>, less<int> > q2;	   //大根堆

void add(int x)
{
   
	if (q1.empty()) //第一个元素入堆
	{
   
		q1.push(x);
		return;
	}
	if (x > q1.top()) //后序元素入堆
		q1.push(x);
	else
		q2.push(x);
	while (q1.size() < q2.size()) //数据转移,保证两个堆的数据数量相差,并且保证小顶堆的堆顶是中位数
	{
   
		q1.push(q2.top());
		q2.pop();
	}
	while (q1.size() > q2.size() + 1) //同上
	{
   
		q2.push(q1.top());
		q1.pop();
	}
}
int main()
{
   
	int t, cnt, n, x;
	scanf("%d", &t);
	while (t--)
	{
   
		while (q1.size()) //清空堆
			q1.pop();
		while (q2.size()) //同上
			q2.pop();
		p.clear(); //多组记得清空一下
		scanf("%d%d", &cnt, &n);
		for (int i = 1; i <= n; ++i)
		{
   
			scanf("%d", &x);
			add(x);
			if (i & 1) //如果是奇数那么就存入数据
				p.push_back(q1.top());
		}
		printf("%d %d\n", cnt, (n + 1) >> 1); //中位数个数 +1在偶数的时候不影响
		for (int i = 0; i < p.size(); ++i)	  //vector数组存数据的时候是从下标为0的时候开始存储的
		{
   
			printf("%d", p[i]);
			if ((i + 1) % 10 == 0)
				puts("");
			else
				printf(" ");
		}
		if (p.size() % 10)
			puts(""); //注意特判最后是否有数据,有才回车。
	}
	return 0;
}
七月已至,年已过半,仲夏来临,愿所有的美好如期而至,所有的不好随风飘散~
全部评论

相关推荐

点赞 评论 收藏
分享
10-28 14:42
门头沟学院 Java
Charles16:你去干的绝对是运维或dba
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务