7-1 寻找大富翁(PTA - 数据结构)
胡润研究院的调查显示,截至2017年底,中国个人资产超过1亿元的高净值人群达15万人。假设给出N个人的个人资产值,请快速找出资产排前M位的大富翁。
·
胡润研究院的调查显示,截至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
提交结果:
思路分析:
乍一看输出前几个最大的,一脑子都是堆排序,结果排序完了,发现打印排好序的前几个元素也行。
要注意的时,m和n的大小问题,整道题难度倒也不大。
代码:
//
// Created by DDD on 2023/12/22.
//
#include <stdio.h>
void Swap(int* pa, int* pb)
{
int tmp = *pa;
*pa = *pb;
*pb = tmp;
}
void AdjustDown(int* arr, int n, int root)
{
int parent = root;
int child = parent * 2 + 1;
while (child < n)
{
if (child + 1 < n && arr[child + 1] < arr[child])
{
child = child + 1;
}
if (arr[child] < arr[parent])
{
Swap(&arr[child], &arr[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
void PrintArray(int* arr, int n)
{
int i;
for (i = 0; i < n; i++)
{
if(i == n-1)
printf("%d", arr[i]);
else printf("%d ", arr[i]);
}
}
void HeapSort(int* arr, int n)
{
int i = 0;
for (i = (n - 1 - 1) / 2; i >= 0; i--){
AdjustDown(arr, n, i);
}
int end = n - 1;
while (end > 0){
Swap(&arr[0], &arr[end]);
AdjustDown(arr, end, 0);
--end;
}
}
int main(){
int n ,m;
scanf("%d %d",&n,&m);
int rich[n];
for (int i = 0; i < n; ++i) {
scanf("%d",&rich[i]);
}
HeapSort(rich, n);
if(n<m)
PrintArray(rich, n);
else PrintArray(rich, m);
}

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐
所有评论(0)