算法设计与分析 李春葆 1_5_2 求无序序列中第k小的元素

题目要求

编写一个实验程序，利用priority_queue（优先队列）求出一个无序整数序列中第k小的元素。

题目分析

1.先请求用户输入数据的个数n。
2.然后开辟一个数组空间用来存储用户输入的数据，然后要求用户输入数据。
3.然后建一个priority_queue（优先队列）来存储那些数组数据。
4.然后请求用户要求输出第k个的k为多少。
5.因为优先队列是从大到小排序，因此第k个小的数据就是第 n-k+1 的数据。

代码实现

#include <iostream>
#include <queue>
using namespace std; 
/*
题目：求无序序列中第k小的元素。 
要求：编写一个实验程序，利用priority_queue（优先队列）求出一个无序整数序列中第k小的元素。 
	  
解题思路：
	1.先请求用户输入数据的个数n。 
	2.然后开辟一个数组空间用来存储用户输入的数据，然后要求用户输入数据。 
	3.然后建一个priority_queue（优先队列）来存储那些数组数据。
	4.然后请求用户要求输出第k个的k为多少。
	5.因为优先队列是从大到小排序，因此第k个小的数据就是第 n-k+1 的数据。 
*/ 
void Pq_KSmall(int array[],int n,int k){		//n为数据数，k为第k小的数据 
	int i,j = 0;
	priority_queue<int,vector<int>,greater<int> > pq;	//为升序排序的优先队列 
	for(i = 0; i<n; i++){				//向队列中压入元素 
		pq.push(array[i]);
	}
	for(i = 1; i<=k; i++){
		j = pq.top();					//利用top函数，返回队头元素，重复向j赋值覆盖，直到第k个元素 
		pq.pop();						//出队操作 
	}
	cout << "第k小的数据为：" << j << endl;
} 

int main(){
	int n,k; 
	cout << "请输入你想记录的数据的个数：";
	cin >> n;
	int array[n];
	for(int i=0; i<n; i++){
		cout << "请输入第"<<i+1<<"个数据：";
		cin >> array[i];
	}
	cout << "请输入你想选出的k：";
	cin >> k;
	Pq_KSmall(array,n,k);
	return 0;
}