浙江工商大学22加试（oj复试）寻找前M个整数

寻找前M个整数

题目描述
给定N个整数，请快速找出排前M位的整数。
输入格式
输入首先给出两个正整数N（N≤1e6）和M（M≤100），其中N为整数个数，M为需要找出的整数个数。接下来一行给出N个整数（所有整数均为小于等于1e18的正整数），数字间以空格分隔。
输出格式
以非递增顺序输出一行前M位的整数，数字间以空格分隔。若N≤M，则输出N个整数的非递增序列，数字间以空格分隔。
样例输入

9 4
100 110 99 105 104 111 101 102 103

样例输出

111 110 105 104

提示/说明
本题可能有一百万个数据输入，请注意算法的时间复杂度。

思路：

寻找最大的m个数有序输出，提示本题有很多数据，那么简单的排序、冒泡排序、sort函数（快排）就过不了。这里的话用小根堆，顾名思义，最小的数在最上层，每次输入一个数，和最上层比较，只要大于最上层的那么就可以放入堆进来，然后进行调整，这样能保证堆里的数是最小的m个，但是并不保证有序输出，所以再用sort函数排序一下。

#include<cstdio>
#include<iostream>
#include<algorithm>
#define int long long 
using namespace std;
int heap[1000001],arr[1000001];


bool cmp(int a,int b){
	if(a>b)  return true;
	else return false;
}

main(){
	int n,m;
	cin>>n>>m;
	for(int i=0;i<m;i++) heap[i]=-1;
	for(int i=0;i<n;i++){
		cin>>arr[i];	//这里用arr数组纯纯冗余 不想改了 
		//建立小根堆 
		if(heap[0]<arr[i]){
			int p=0;
			heap[0]=arr[i];
			while(p<m){
				int q=2*p+1;
				if(q>m) break;		//只需要保存m个 如果大于m，就不需要往下放了 
				if((q<m-1)&&heap[q]>heap[q+1]){//如果同一层有两个结点，用更小的节点去替换 
					q+=1;
				}
				if(heap[p]>heap[q]){	//越小的数据越往上移动 
					swap(heap[p],heap[q]);
					p=q;	
				}else{
					break;
				}
			}
		}
	}
	sort(heap,heap+m,cmp);	//小根堆不保证有序 
	for(int i=0;i<m;i++){
		cout<<heap[i];
		if(i<m-1) cout<<" ";
	}
} ```