堆排序(优化版)

数据结构实验之排序四:寻找大富翁
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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值