day2:53 最小的k个数(大根堆) and 703. 数据流中的第 K 大元素(小根堆)

问题描述: 输入n个整数,找出其中最小的k个数。

  • 数据保证k一定小于等于输入数组的长度;
  • 输出数组内元素请按从小到大顺序排序;
    样例
  • 输入:[1,2,3,4,5,6,7,8] , k=4
  • 输出:[1,2,3,4]

好久不写c++ 代码了,介绍一个C++11上的新增功能auto关键字(使用auto可以简化for循环中的代码书写),在使用DevC++5.11编译运行时需要更改设置,否则无法通过编译在Tools–>Compiler options–>seetting–>code generation,如下图中设置编译环境,修改后即可。
在这里插入图片描述
应用大根堆实现此问题,c++中自带了大根堆C++ 标准库中的堆(heap):

  • make_heap: 根据指定的迭代器区间以及一个可选的比较函数,来创建一个heap. O(N)
  • push_heap: 把指定区间的最后一个元素插入到heap中. O(logN)
  • pop_heap: 弹出heap顶元素, 将其放置于区间末尾. O(logN)
  • sort_heap:堆排序算法,通常通过反复调用pop_heap来实现. N*O(logN)
    C++11加入了两个新成员:
  • is_heap: 判断给定区间是否是一个heap. O(N)
  • is_heap_until: 找出区间中第一个不满足heap条件的位置. O(N)

本次使用的是优先队列本质是也是使用堆实现的。链接: https://blog.csdn.net/weixin_36888577/article/details/79937886

class Solution {
public:
    vector<int> getLeastNumbers_Solution(vector<int> input, int k) {
        priority_queue<int>  heap;
		for(auto x:input )
		{
			heap.push(x);
			if(heap.size()>k) heap.pop();
		 } 
		 vector<int> res;
		 while(heap.size()) res.push_back(heap.top()),heap.pop();
    
    	reverse(res.begin(),res.end());
		return res;
	}
};

由于优先队列是大根堆,考虑到本文是做从小到大排序,并且只要K个数字。如果heap.size()>k,完全可以删除队头元素。剩下的K个元素即为要得到的元素,将元素依次读取队头,并保存在vector这一容器中,之后对该序列进行转置输出即为所求。

703. 数据流中的第 K 大元素
设计一个找到数据流中第 k 大元素的类(class)。注意是排序后的第 k 大元素,不是第 k 个不同的元素。请实现 KthLargest 类:

KthLargest(int k, int[] nums) 使用整数 k 和整数流 nums 初始化对象。
int add(int val) 将 val 插入数据流 nums 后,返回当前数据流中第 k 大的元素。

示例:
输入:
[“KthLargest”, “add”, “add”, “add”, “add”, “add”]
[[3, [4, 5, 8, 2]], [3], [5], [10], [9], [4]]
输出:
[null, 4, 5, 5, 8, 8]

解释:
KthLargest kthLargest = new KthLargest(3, [4, 5, 8, 2]);
kthLargest.add(3); // return 4
kthLargest.add(5); // return 5
kthLargest.add(10); // return 5
kthLargest.add(9); // return 8
kthLargest.add(4); // return 8

class KthLargest {
public:
    priority_queue<int,vector<int>,greater<int>> q;
    int knum;
    KthLargest(int k, vector<int>& nums) {
        knum=k;
        for(auto x:nums) add(x);
    }
    
    int add(int val) {
        q.push(val);
        if(q.size()>knum) q.pop();
        return q.top();
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值