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
7 1 3 5 7 9 11 6
Sample Output1
1 3 5
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;
}
else
break;
}
else //小根堆
{
if (j < a[0] && a[j+1] < a[j]) //小根堆要调整则要找小的。
j++;
if (x > a[j])
{
a[i] = a[j];
i = j;
j = i*2;
}
else
break;
}
}
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;
}