数据结构实验之排序四:寻找大富翁
TimeLimit: 200MS Memory Limit: 512KB
ProblemDescription
2015胡润全球财富榜调查显示,个人资产在1000万以上的高净值人群达到200万人,假设给出N个人的个人资产值,请你快速找出排前M Input
首先输入两个正整数N(N ≤ 10^6)和M(M≤10),其中N为总人数,M为需要找出的大富翁数目,接下来给出N个人的个人资产,以万元为单位,个人资产数字为正整数,数字间以空格分隔。
Output
一行数据,按降序输出资产排前M
ExampleInput
63
126 56 23 188 60
ExampleOutput
18860 56
Hint
请用堆排序完成。
Author
xam
#include <stdio.h> //头文件只有一个防超时
int h[1000002];//用来存放堆的数组
int m; //堆的大小
//交换函数,用来交换堆中的两个元素的值
void Swap(int x, int y)
{
int t;
t = h[x];
h[x] = h[y];
h[y] = t;
}
//向下调整函数
void Siftdown(int i) //传入一个需要向下调整的节点的编号i,这里传入1,即
//从堆顶开始向下调整;
{
int t, flag=0; //flag用来标记是否需要继续向下调整
//当i结点有孩子(其实是至少有左孩子)并且有需要继续调整的时候循环就执行
while(i*2 <= m && flag == 0)
{
//首先判断它和左孩子的关系,并用t记录值较小的结点的编号
if(h[i] > h[i*2])
t = i*2;
else
t = i;
//如果它有右孩子,再对右孩子进行讨论
if(i*2+1 <= m)
{
//假如右孩子的值更小,更新较小的节点编号
if(h[t] > h[i*2+1])
t = i*2+1;
}
//如果发现最小的结点编号不是自己,说明子节点中有比父节点更小的
if(t != i)
{
Swap(t, i); //交换他们
i = t; //更新i为刚才与它交换的孩子结点的编号,便于接下来继续向下调整
}
else
flag = 1; //否则说明当前的父节点已经比两个子节点都要小了,不需要再进行调整了
}
}
void heapsort() //堆排序
{
while(m > 1)
{
Swap(m, 1);
m--;
Siftdown(m);
}
}
int main()
{
int num, n, i, j;
scanf("%d%d", &num, &m);
for(i = 1; i <= m; i++)
scanf("%d", &h[i]);
n = m;
for(i = m/2; i >= 1; i--) //建m个元素的最小堆
Siftdown(i);
for(i = m+1; i <= num; i++) //然后从第m+1个元素开始,与堆顶元素比较,
//如果比堆顶的元素小,那这个数就不要了,
{
int x;
scanf("%d", &x);
if(x > h[1])
{
//如果比堆顶的元素大,那就舍弃当前堆顶而
//将这个数作为新的堆顶,并维护最小堆的性质
h[1] = x;
for(j = m/2; j >= 1; j--)//可以改进把此行删除
Siftdown(j);
}
}
heapsort(); //堆排序
for(i = 1; i <= n; i++)
if(i != n)
printf("%d ", h[i]);
else
printf("%d\n", h[i]);
return 0;
}
/*求前k个最大的数则建立k个数的最小堆
求前k个最小的数则建立k个数的最大堆
算法描述:我们首先取这N个元素中的前K个元素来建立一个由K个元素组成的小(大)顶堆,这样堆顶元素便是当前已读取元素中的第K大(小)者;然后,依次读取剩下的N-K个元素,而对于其中的每个元素x,若x大于(小于)当前的堆顶元素,则将堆顶元素替换为x,并自堆顶至向下调整堆;这样,在读取完所有元素后,堆中的K个元素即为这N个数中的前K个最大(小)元素,同时堆顶元素为这N个数中的第K大(小)元素。*/
/***************************************************
User name: jk160505徐红博
Result: Accepted
Take time: 192ms
Take Memory: 104KB
Submit time: 2017-02-23 19:12:43
****************************************************/