数据结构实验之排序四:寻找大富翁
Time Limit: 200 ms Memory Limit: 512 KiB
Problem Description
2015胡润全球财富榜调查显示,个人资产在1000万以上的高净值人群达到200万人,假设给出N个人的个人资产值,请你快速找出排前M位的大富翁。
Input
首先输入两个正整数N( N ≤ 10^6)和M(M ≤ 10),其中N为总人数,M为需要找出的大富翁数目,接下来给出N个人的个人资产,以万元为单位,个人资产数字为正整数,数字间以空格分隔。
Output
一行数据,按降序输出资产排前M位的大富翁的个人资产值,数字间以空格分隔,行末不得有多余空格。
Sample Input
6 3
12 6 56 23 188 60
Sample Output
188 60 56
Hint
请用堆排序完成。
Source
xam
思路:
1.堆排序的思路就是先按照降序升序建立最大堆最小堆(建堆)
把堆顶元素和最后一个元素进行交换,并且向下调整,依次进行n次;(堆排序)
最后输出;
2.向上调整就是一个建堆的过程,附上代码;
1:
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
int a[1000001];
using namespace std;
void siftdown(int i,int n)//适合删除数据
{
int t,f=0;
while(i*2<=n&&f==0)
{
if(a[i]>a[i*2])
t=2*i;
else t=i;
if(i*2+1<=n)
{
if(a[t]>a[i*2+1])
{
t=i*2+1;
}
}
if(t!=i)
{
swap(a[i],a[t]);
i=t;
}
else f=1;
}
}
void siftup(int i)//适合插入数据;
{
int f=0;
if(i==1)return ;
while(i!=1&&f==0)
{
if(a[i]<a[i/2])
swap(a[i],a[i/2]);
else f=1;
i=i/2;
}
}
void create(int n)
{
for(int i=n/2;i>=1;i--)
{
siftdown(i,n);
}
}
void heapsort(int n)
{
while(n>1)
{
swap(a[1],a[n]);
n--;
siftdown(1,n);//只需调整一次就可以,因为都是已经调整好了的,当把第一个和最后一个交换之后,向下调整时会把次小的拿到堆顶,最大的那个元素归位;重复;
}
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)
{
scanf("%d",&a[i]);
}
create(m);//一定要建立最小堆,因为不建立就不能保证第一个是最小的,以后更新的时候也没有办法更新;
for(int i=m+1;i<=n;i++)
{
int t;
scanf("%d",&t);
if(t>a[1])
{
a[1]=t;
siftdown(1,m);//这个只是把M个数里面的最小的用大的替换,然后次小的再上到堆顶,依次这样不断更新小的数,最后得到前M大的数;
}
}
heapsort(m);
for(int i=1;i<=m;i++)
{
if(i==1)printf("%d",a[i]);
else printf(" %d",a[i]);
}
printf("\n");
return 0;
}
2.
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
int a[1000001];
using namespace std;
void siftdown(int i,int n)//适合删除数据
{
int t,f=0;
while(i*2<=n&&f==0)
{
if(a[i]>a[i*2])
t=2*i;
else t=i;
if(i*2+1<=n)
{
if(a[t]>a[i*2+1])
{
t=i*2+1;
}
}
if(t!=i)
{
swap(a[i],a[t]);
i=t;
}
else f=1;
}
}
void siftup(int i)//适合插入数据;
{
int f=0;
if(i==1)return ;
while(i!=1&&f==0)
{
if(a[i]<a[i/2])
swap(a[i],a[i/2]);
else f=1;
i=i/2;
}
}
//void create(int n)
//{
// for(int i=n/2;i>=1;i--)
// {
// siftdown(i,n);
// }
//}
void heapsort(int n)
{
while(n>1)
{
swap(a[1],a[n]);
n--;
siftdown(1,n);
}
}
int main()
{
int n,m,p=1;
scanf("%d%d",&n,&m);
scanf("%d",&a[p++]);
for(int i=2;i<=m;i++)
{
int t;
scanf("%d",&t);
a[p++]=t;
siftup(p-1);//这里的建堆利用了向上调整函数,没有用向下调整,也是正确的;
}
// create(m);
for(int i=m+1;i<=n;i++)
{
int t;
scanf("%d",&t);
if(t>a[1])
{
a[1]=t;
siftdown(1,m);//删除用向下调整;
}
}
heapsort(m);
for(int i=1;i<=m;i++)
{
if(i==1)printf("%d",a[i]);
else printf(" %d",a[i]);
}
printf("\n");
return 0;
}