C++选择问题算法

C++选择问题算法实现

原创于 2017-01-07 21:11:01 发布 · 1.5k 阅读

0 ·

CC 4.0 BY-SA版权

C++数据结构专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种基于C++的选择问题算法实现方法，通过模板实现了泛型编程，能够找到未排序数组中第k大的元素。该算法首先对前k个元素进行排序，然后遍历剩余元素并维护一个大小为k的候选集合。

最近在学习数据结构和算法，这里贴上选择问题的c++实现，函数是基于模板的，通用性较高。

#include<iostream>
#include<vector>
#include<algorithm>
#include <functional>
using namespace std;

template<typename T>
T select(const vector<T>&vt, int k)//总元素的个数用vt.size()表示
{
	typename vector<T>VT(&vt[0], &vt[k]);
	sort(VT.begin(), VT.end(), greater<T>());
	for (size_t i = k;i < vt.size();i++)
	{
		if((VT[k-1]) < vt[i])
			VT[k-1] =(vt[i]);
		sort(VT.begin(), VT.end(), greater<T>());
	}
	return VT[k-1];
}

int main()
{
	int arr[] = { 1,3,2,5,4,8,6 };
	vector<int>vt(arr,arr+7);
	vt.reserve(7);
	cout << select<int>(vt, 3);
	return 0;
}