最小的K个数

题目：最小的K个数

题目描述

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

思路

①用优先队列（最大堆），最大值在queue.top(), 先进k个，然后后面的数据和queue.top比较,小的留下

#include<bits/stdc++.h> 
#include<queue>
#include<vector>
using namespace std;
class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res;
        if(k<1||input.size()<k) return res;
        priority_queue<int, vector<int>, less<int> >q; //less代表单调减，大的在queue前 
        int i=0; 
		while(i<k) q.push(input[i++]); //插入k个进队列中 
		for( ;i<input.size();i++){
        	if(input[i]<q.top()){ // 第i个小于队列中最大值 
        		q.pop();
        		q.push(input[i]);
			}
		}
		while(!q.empty()){
			res.insert(res.begin(), q.top());//插入到开头 
			q.pop();
		}
        return res;
    }
};
int main(){
	Solution s;
	vector<int> v = {4,5,1,6,2,7,3,8};
	v = s.GetLeastNumbers_Solution(v,4);
	auto it = v.begin();
	while(it!=v.end()){
		cout<<*it<<" ";
		it++;
	} 
	return 0;
}

 

 

 

②排序，然后返回前k个

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res;
        if(k<1||input.size()<k) return res;
        sort(input.begin(), input.end());
        res.assign(input.begin(),input.begin()+k);
        return res;
    }
};

 

笔记

#include<queue>

priority_queue<int, vector<int>, less<int> > q; // less代表队列从大到小排序

priority_queue<int, vector<int>, greater<int> > p; // greater 代表队列从小到大排序 

q.push(val) 插入队列中

q.pop()   弹出queue头元素

q.top()   访问queue头元素

q.empty() q.size() 返回是否空，和大小

 

heap不属于STL，尽可能用priority_queue代替

#include <algorithm> // make_heap(), pop_heap(), push_heap()

v = {4,5,1,6,2,7,3,8};
make_heap(v.begin(), v.end(), less<int>()); //最大值就在v[0],默认最大堆
make_heap(v.begin(), v.end(), greater<int>()); //最小值就在v[0]，最小堆 
    
pop_heap(v.begin(), v.end(), greater<int>());//先pop,然后在容器中删除
v.pop_back();
 
v.push_back(100);//先在容器中加入，再push
push_heap(v.begin(), v.end(), greater<int>());
 
sort_heap(v.begin(), v.end());