【u119】中位数

Time Limit: 1 second
Memory Limit: 128 MB

【问题描述】

给出一个长度为N的非负整数序列A[i]，对于所有1 ≤ k ≤ (N + 1) / 2，输出A[1], A[2], …, 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[2], …, A[2i – 1]的中位数。

【数据规模】

对于20%的数据，N ≤ 100； 对于40%的数据，N ≤ 3000； 对于100%的数据，N ≤ 100000。

Sample Input1

71 3 5 7 9 11 6

Sample Output1

135
6
【题解】
维护一个大根堆和一个小根堆。
一开始把第一二元素加入到大根堆的第一个位置。然后直接输出第一个元素。
对于之后输入的元素。进行判断。如果大于大根堆的根，则放到小根堆。小于或等于则放到大根堆。
然后放完之后。进行一次判断。看一下这两个堆它们的差的绝对值是否会大于2.如果大于2的话。就把数字
多的那个堆中的根放到另外一个堆中。这样的判断也做完之后。就可以输出中位数了。比如当前i=3，则中
位数的位置就在2，先判断一下2是否等于大根堆的大小，如果是就输出大根堆的根，否则输出小根堆的根。
难点的话在堆的操作上。多熟悉就懂了。
【代码】
#include <cstdio>int dagendui[100001],xiaogendui[100001],n,pos; //拼音分别对应了大根堆和小根堆 void up_tiaozheng(int a[100001],int p,int what) //往上调整。 what对应了a堆是什么堆 { //0为大根堆 1为小根堆 int x = a[p]; //获取需要调整的数字 int i = p,j = p/2; //往上调整 所以是除2表示指向其父亲节点 //printf("p=%d\n",p);while (j > 0) //如果没有越过树的范围 {if (what ==0) //大根堆 的调整方法 {if (x > a[j]) //如果这个值比父亲节点大。那么就不符合大根堆的定义了。 { //往上走 a[i] = a[j];i = j;j = i/2;}else //否则就找到了一个合适的位置 直接结束即可。 break;}else //小根堆 {if (x <a[j]) //如果父亲节点比儿子节点大。则不符合小根堆的定义。 { //同样进行调整 a[i] = a[j];i = j;j = i/2;}else //如果找到合适的位置就直接结束调整 break;}}a[i] = x;pos = i; //要记录这个数字最后放到了哪里。 }void down_tiaozheng(int a[100001],int p,int what) //往下调整。 {int x = a[p]; //获取需要调整的数字。 int i = p,j = p*2;while (j <=a[0] ){if (what ==0) //大根堆 {if (j < a[0] && a[j+1] > a[j]) //大根堆往下调整的话，儿子要找大的。 j++;if (x < a[j]){a[i] = a[j];i = j;j = i*2;}elsebreak;}else //小根堆 {if (j < a[0] && a[j+1] < a[j]) //小根堆要调整则要找小的。 j++;if (x > a[j]){a[i] = a[j];i = j;j = i*2;}elsebreak;}}a[i] = x;}void input_data(){scanf("%d",&n);scanf("%d",&dagendui[1]); //先把第一个元素放到大根堆的根节点。 dagendui[0] = 1;printf("%d\n",dagendui[1]);for (int i = 2;i <= n;i++){int dd;scanf("%d",&dd);//输入的数据和大根堆的根节点比较  if (dd > dagendui[1]) //根据比较的结果放到大根堆或小根堆。 {xiaogendui[0]++;xiaogendui[xiaogendui[0]]=dd;up_tiaozheng(xiaogendui,xiaogendui[0],1); //放到末尾要先向上调整再向下调整 down_tiaozheng(xiaogendui,pos,1);}else{dagendui[0]++;dagendui[dagendui[0]] = dd;up_tiaozheng(dagendui,dagendui[0],0);down_tiaozheng(dagendui,pos,0);//同理 }if (dagendui[0] > xiaogendui[0]) //如果它们的大小之差的绝对值大于2则需要调整 {if ((dagendui[0] - xiaogendui[0]) > 2) //大根堆数字比较多 { //就把大根堆的根节点放到小根堆中去 xiaogendui[0]++;xiaogendui[xiaogendui[0]] = dagendui[1];up_tiaozheng(xiaogendui,xiaogendui[0],1);down_tiaozheng(xiaogendui,pos,1);dagendui[1] = dagendui[dagendui[0]];dagendui[0]--;down_tiaozheng(dagendui,1,0);}}else if (dagendui[0] < xiaogendui[0]){if ((xiaogendui[0]-dagendui[0]) > 2){ //如果小根堆中的数字更多。则把小根堆的根节点放到大根堆中去。 dagendui[0]++;dagendui[dagendui[0]] = xiaogendui[1];up_tiaozheng(dagendui,dagendui[0],0);down_tiaozheng(dagendui,pos,0);xiaogendui[1] = xiaogendui[xiaogendui[0]];xiaogendui[0]--;down_tiaozheng(xiaogendui,1,1);}}if ( (i%2)==1) //如果是奇数 则输出大根堆的根节点或小根堆的根节点。 {int tt = (i+1)/2;if (tt ==dagendui[0]) //大根堆的大小和所需要输出的第tt个数字相同就可以直接输出根节点 printf("%d\n",dagendui[1]);else //否则的话就是放在小根堆的根节点了。 printf("%d\n",xiaogendui[1]);}}}int main(){//freopen("F:\\rush.txt","r",stdin);input_data();return 0;}