数据结构实验之排序四:寻找大富翁
Time Limit: 150ms Memory limit: 512K 有疑问?点这里^_^
题目描述
2015胡润全球财富榜调查显示,个人资产在1000万以上的高净值人群达到200万人,假设给出N个人的个人资产值,请你快速找出排前M位的大富翁。
输入
首先输入两个正整数N( N ≤ 10^6)和M(M ≤ 10),其中N为总人数,M为需要找出的大富翁数目,接下来给出N个人的个人资产,以万元为单位,个人资产数字为正整数,数字间以空格分隔。
输出
一行数据,按降序输出资产排前M位的大富翁的个人资产值,数字间以空格分隔,行末不得有多余空格。
示例输入
6 3 12 6 56 23 188 60
示例输出
188 60 56
提示
请用堆排序完成。
来源
xam
示例程序
#include<bits/stdc++.h>
using namespace std;
void heap(int a[],int i,int size)
{
int temp;
if(i<=size/2)
{
int left=2*i;
int right=2*i+1;
int root=i;
if(left<=size&&a[root]>a[left])
{
temp=a[root];
a[root]=a[left];
a[left]=temp;
root=left;
}
if(right<=size&&a[root]>a[right])
{
temp=a[root];
a[root]=a[right];
a[right]=temp;
root=right;
}
if(root!=i)
heap(a,root,size);
}
}
int main()
{
int i,n,m,x;
int a[11];
scanf("%d%d",&n,&m);
for(int i=1; i<=m; ++i)
scanf("%d",&a[i]);
for(int i=m/2; i>=1; --i)
heap(a,i,m);
for(int j=m+1; j<=n; j++)
{
scanf("%d",&x);
if(x>a[1])
{
a[1]=x;
heap(a,1,m);
}
}
for(int i=m; i>=1; --i)
{
swap(a[1],a[i]);
heap(a,1,i-1);
}
for(i=1; i<m; i++)
printf("%d ",a[i]);
printf("%d\n",a[m]);
}
//课本上的方法
#include <bits/stdc++.h>
using namespace std;
int a[11];
void heapjust(int s,int m) //把a[]建成小顶堆
{
int t=a[s];
for(int i=2*s; i<=m; i*=2)
{
if(i<m&&a[i]>a[i+1]) //记录两个孩子节点较小的位置
i++;
if(t<a[i])
break;
a[s]=a[i];
s=i;
}
a[s]=t;
}
int main()
{
int n,m,i,j;
while(~scanf("%d %d",&n,&m))
{
for(i=1; i<=m; i++) //先建成m个数的小顶堆
scanf("%d",&a[i]);
for(i=m/2; i>0; i--) //从m/2开始比较
heapjust(i,m);
for(i=m+1; i<=n; i++)
{
int x;
scanf("%d",&x);
if(x<=a[1])
continue;
a[1]=x; //更新小顶堆
for(j=m/2; j>0; j--)
heapjust(j,m);
}
for(i=m;i>1;i--) //最后堆排序
{
int t=a[1];
a[1]=a[i];
a[i]=t;
heapjust(1,i-1);
}
for(i=1; i<=m; i++)
printf("%d%c",a[i],i==m?'\n':' ');
}
}