PAT甲级1029 Median (25 分)题解 优先队列解决爆内存

\quad 这个题目的是找出两个序列的中位数，给的内存为1.5M，单个序列大小为$2\ast 10^5$，int型。若将两个序列存储起来，则需要存储空间大小为$4\ast 10^5\ast 4Bytes=1.6M$，会爆内存。设两个序列长度为num1,num2，那么排好序后中位数索引为$mid=(nu{m}_1+nu{m}_2)/2$。因此我们只需要维护一个大小为mid的优先队列，里面存放最小的mid个元素即可。具体流程如下：

• 将第一个序列元素入队，若num1个数大于mid则需要删除多余部分，若小于mid则需要继续读取第二个序列中元素，直到维持队列中有mid个元素
• 序列2中读入一个元素temp，判断temp是否小于当前队列中最大值，若小于，则入队并将队中最大值出队
• 若temp大于当前队列最大值，因为序列2有序，直接退出循环输出答案即可，答案就是队列中最大值，即队首元素
  

#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;

priority_queue<int> q;
int main(int argc, char const *argv[])
{
	ios::sync_with_stdio(false);
	int num1, num2, temp;
	cin >> num1;
	for (int i = 0; i < num1; ++i){
		cin >> temp;
		q.push(temp);
	}
	cin >> num2;
	int mid = (num1+num2+1)/2; // 中间值索引为mid
	while(q.size()>mid) q.pop(); // 如果第1个序列元素大于mid，则删除多余部分
	for (int i = 0; i < num2; ++i){
		cin >> temp;
		if(q.size()<mid) q.push(temp); // 若优先队列中元素个数不足mid，先补齐为mid个元素
		else if(temp<q.top()){  // 若q中最大的数小于当前读入的数，则将次数入队并将最大值出队，维持大小不变
			q.push(temp); 
			q.pop();
		}
		else break;
	}
	cout << q.top();
	return 0;
}