洛谷在线测试P1168_中位数

来源:互联网 发布:淘宝网首页小音箱 编辑:程序博客网 时间:2024/05/29 02:55
/*  Name: P1168_中位数  Copyright:   Author: 巧若拙   Date: 01-03-17 07:37  Description:   P1168 中位数题目描述给出一个长度为N的非负整数序列A[i],对于所有1 ≤ k ≤ (N + 1) / 2,输出A[1], A[3], …, A[2k - 1]的中位数。即前1,3,5,……个数的中位数。输入输出格式输入格式:输入文件median.in的第1行为一个正整数N,表示了序列长度。第2行包含N个非负整数A[i] (A[i] ≤ 10^9)。输出格式:输出文件median.out包含(N + 1) / 2行,第i行为A[1], A[3], …, A[2i – 1]的中位数。输入输出样例输入样例#1:71 3 5 7 9 11 6输出样例#1:1356说明对于20%的数据,N ≤ 100;对于40%的数据,N ≤ 3000;对于100%的数据,N ≤ 100000。算法思路:左右分别构造一个最大堆L和最小堆R,把中位数作为最大堆的堆顶元素。先读入第一个数,其即第一个中位数L.top(); 之后每次读入两个数,若小于L.top(),插入到最大堆,否则插入到最小堆;若读入的两个数是插入到同一个堆中,则需要删除该堆的堆顶元素,插入到另一个堆中,以确保两个堆的结点数满足maxHeapSize-minHeapSize==1。 注意:要小心当n为偶数的情形,最后一个数字是没有用的,所以只处理前2*i-1个数字 分点信息(鼠标移到方块上有详细信息)#1AC2ms/1378kB #2AC13ms/16421kB #3AC272ms/1687kB #4AC408ms/1710kB #5AC449ms/1664kB*/#include<iostream>#include<algorithm>#include<math.h>#include<queue>using namespace std;int main(){ priority_queue<int> L; //左侧的最大堆Lpriority_queue<int, vector<int>, greater<int> > R; //右侧的最小堆R int n, x;cin >> n;cin >> x;L.push(x); //把中位数存放在左侧的最大堆L中 cout << L.top() << endl;//输出中位数 for (int i = 1; i <= ((n-1)>>1); i++)//只处理前2*i-1个数字 {int l_r = 0; //用来表示新读入的两个数是插入到了左侧还是右侧,左侧--,右侧++  for(int j=0; j<2; j++) { cin >> x; if(x < L.top()) {L.push(x);l_r--;        }else{R.push(x);l_r++;        }} if(l_r < 0) //两个数都插入到了L中{R.push(L.top());L.pop();}else if(l_r > 0) //两个数都插入到了R中{L.push(R.top());R.pop();}cout << L.top() << endl;//输出中位数}//    system("pause");   return 0;}

0 0
原创粉丝点击