priority_queue、multiset的简单使用

最新推荐文章于 2022-03-06 09:45:47 发布

翻译最新推荐文章于 2022-03-06 09:45:47 发布 · 361 阅读

C++ 专栏收录该内容

53 篇文章

订阅专栏

本文介绍了一种使用优先队列和多集合数据结构来找出数组中最小的k个数的方法。通过实例展示了如何利用这些数据结构高效地解决问题，对于理解和应用数据结构与算法具有很好的指导意义。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution {
public:
	vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
		int size = input.size();
		vector<int> ret;
        
        if (size == 0 || k <= 0 || k > size)
            return ret;
        
		priority_queue<int> q;
		for (int i = 0; i < size; i++) {
			if (q.size() < k)
				q.push(input[i]);
			else {
				if (input[i] < q.top()) {
					q.pop();
					q.push(input[i]);
				}
			}
		}
 
		while (!q.empty()) {
			ret.push_back(q.top());
			q.pop();
		}
 
		reverse(ret.begin(), ret.end());
 
		return ret;
	}
};

class Solution{
public:
	vector<int> GetLeastNumbers(vector<int> input, int k){
		if(input.size() < k || k < 0)
			return vector<int>();
		
		multiset<int, greater<int>> leastNumbers;
		vector<int> ::const_iterator iter = input.begin();
		for(; iter != input.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);
				}
			}
		}
		return vector<int>(leastNumbers.begin(), leastNumbers.end());
	}
	
};