PTA:7-13 寻找大富翁(C语言)

 胡润研究院的调查显示,截至2017年底,中国个人资产超过1亿元的高净值人群达15万人。假设给出N个人的个人资产值,请快速找出资产排前M位的大富翁。

输入格式:

输入首先给出两个正整数N(≤106)和M(≤10),其中N为总人数,M为需要找出的大富翁数;接下来一行给出N个人的个人资产值,以百万元为单位,为不超过长整型范围的整数。数字间以空格分隔。

输出格式:

在一行内按非递增顺序输出资产排前M位的大富翁的个人资产值。数字间以空格分隔,但结尾不得有多余空格。

输入样例:

8 3
8 12 7 3 20 9 5 18

输出样例:

20 18 12

参考代码:

/**
 * 7-13 寻找大富翁
 *  堆排序和选择排序
 */

#include <stdio.h>   //堆排序;  注意:此算法中,下标从1开始

#define max 1000010
int num[max];

void sift(int *num, int low, int high)  //将下标为low位置上的点调到适当位置
{
    int i, j, temp;
    i = low;
    j = 2 * i;   //num[j]是num[i]的左孩子结点;
    temp = num[i];  //待调整的结点
    while (j <= high) {
        if (j < high && num[j] < num[j + 1])   //如果右孩子比较大,则把j指向右孩子,j变为2*i+1;
            ++j;
        if (temp < num[j]) {
            num[i] = num[j];    //将num[j]调整到双亲结点的位置上;
            i = j;   //修改i和j的值,以便继续向下调整;
            j = i * 2;
        } else break;     //调整结束;
    }
    num[i] = temp;   //被调整的结点放入最终位置
}

int main() {
    int n, m, i, temp, count = 0;
    scanf("%d%d", &n, &m);
    for (i = 1; i <= n; i++)
        scanf("%d", &num[i]);
    if (n < m) m = n;   //注意,有一个测试点是n小于m的情况,这时,只用排前n个;
    for (i = n / 2; i >= 1; i--)  //所有结点建立初始堆
        sift(num, i, n);
    for (i = n; i >= 2; i--)   //进行n-1次循环,完成堆排序
    {
        /*以下3句换出了根节点中的关键字,将其放入最终位置*/
        temp = num[1];
        num[1] = num[i];
        num[i] = temp;
        count++;
        if (count == 1)
            printf("%d", num[i]);
        else
            printf(" %d", num[i]);
        if (count == m) break;  //打印前m个;
        sift(num, 1, i - 1);    //减少了1个关键字的无序序列,继续调整。
    }
    if (m == n) printf(" %d", num[1]);  //当n<m的特殊情况下,上面只打印了n~2,还有最后一个下标为1的没有打印,故再打印一个。
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_61228493/article/details/131108469