堆排——資料結構實驗之排序四:尋找大富翁
阿新 • • 發佈:2019-01-13
資料結構實驗之排序四:尋找大富翁
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
請用堆排序完成。
該題是時間限制的很緊,所以使用堆排序,理清思路,慢慢編寫。
為什麼建立大頂堆的時候,從m/2到m也能AC??課本上介紹的是從m/2到1呀!!
#include<bits/stdc++.h> using namespace std; int a[1000001]; void HeapAdjust(int a[], int i, int m) { int t, q; t = i; while(2 * i <= m) { if(a[i] < a[2 * i]) { t = 2 * i; } if(a[t] < a[2 * i + 1] && 2 * i + 1 <= m) { t = 2 * i + 1; } if(t == i) { break; } if(t != i) { q = a[t]; a[t] = a[i]; a[i] = q; i = t; } } } void Heapsort(int a[], int i, int m) { int t; while(m >= 1) { t = a[1]; a[1] = a[m]; a[m] = t; m--; HeapAdjust(a, 1, m); } } int main(void) { int n, m, i, x; scanf("%d %d", &n, &m); for(i = 1; i <= m; i++) { scanf("%d", &a[i]); } for(i = m / 2; i <= m; i++) { HeapAdjust(a, i, m); } Heapsort(a, 1, m); for(i = m + 1; i <= n; i++) { scanf("%d", &x); if(x > a[1]) { a[1] = x; HeapAdjust(a, 1, m); Heapsort(a, 1, m); } } for(i = m; i >= 1; i--) { printf("%d%c", a[i], i == 1 ?'\n':' '); } return 0; }