码迷,mamicode.com
首页 > 编程语言 > 详细

编程算法 - 最小的k个数 红黑树 代码(C++)

时间:2014-10-09 23:56:17      阅读:280      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   color   io   os   使用   ar   for   

最小的k个数 红黑树 代码(C++)


本文地址: http://blog.csdn.net/caroline_wendy


题目: 输入n个整数, 找出当中的最小k个数.


使用红黑树(multiset), 每次替换最大的值, 依次迭代.

时间复杂度: O(nlogk).


代码:

/*
 * main.cpp
 *
 *  Created on: 2014年6月29日
 *      Author: wang
 */

#include <iostream>
#include <vector>
#include <algorithm>
#include <iterator>
#include <set>

using namespace std;

void GetLeastNumbers (const vector<int>& data,
		multiset<int, greater<int> >& leastNumbers,
		std::size_t k)
{
	leastNumbers.clear();

	if (k < 1 || data.size() < k)
		return;

	for (vector<int>::const_iterator iter = data.begin();
			iter != data.end(); ++iter)
	{
		if (leastNumbers.size() < k) {
			leastNumbers.insert(*iter);
		} else {
			multiset<int, greater<int> >::iterator iterGreatest = leastNumbers.begin();

			if (*iter < *iterGreatest) {
				leastNumbers.erase(iterGreatest);
				leastNumbers.insert(*iter);
			}
		}
	}
}

int main(void)
{
    vector<int> input =  {4, 5, 1, 6, 2, 7, 3, 8};
    multiset<int, greater<int> > output;
    GetLeastNumbers (input, output, 4);
    std::copy(output.begin(), output.end(), std::ostream_iterator<int>(std::cout, " "));
    std::cout << std::endl;
    return 0;
}



输出:

4 3 2 1 



bubuko.com,布布扣



编程算法 - 最小的k个数 红黑树 代码(C++)

标签:style   blog   http   color   io   os   使用   ar   for   

原文地址:http://www.cnblogs.com/mengfanrong/p/4014298.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!