leetcode题:703. 数据流中的第K大元素(简单)

2024-03-15 15:38

本文主要是介绍leetcode题:703. 数据流中的第K大元素(简单),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

一、题目描述:703. 数据流中的第K大元素(简单)

设计一个找到数据流中第K大元素的类(class)。注意是排序后的第K大元素,不是第K个不同的元素。

你的 KthLargest 类需要一个同时接收整数 k 和整数数组nums 的构造器,它包含数据流中的初始元素。每次调用 KthLargest.add,返回当前数据流中第K大的元素。

示例:

int k = 3;
int[] arr = [4,5,8,2];
KthLargest kthLargest = new KthLargest(3, arr);
kthLargest.add(3);   // returns 4
kthLargest.add(5);   // returns 5
kthLargest.add(10);  // returns 5
kthLargest.add(9);   // returns 8
kthLargest.add(4);   // returns 8
说明: 
你可以假设 nums 的长度≥ k-1 且k ≥ 1。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/kth-largest-element-in-a-stream
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

二、解题思路

使用最小堆,堆元素数量小于k的时候,val直接入堆,否则当val大于最小堆的堆顶元素的时候,val加入到最小堆,并且删除最小堆堆顶元素。

返回最小堆堆顶元素

三、代码

class rev
{public:bool operator()(const int & left,const int & right) const;
};
bool rev::operator()(const int & left,const int &right) const{return left > right;}
class KthLargest {
public:KthLargest(int k, vector<int>& nums) {size = k;for(int i = 0;i < nums.size();i++){save.insert(nums[i]);if(save.size() > size)save.erase(save.begin());}}multiset<int> save;int size;int add(int val) {if(save.size() >= size && *(save.begin()) < val){save.erase(save.begin());save.insert(val);}else if(save.size() < size){save.insert(val);}return *(save.begin());}};/*** Your KthLargest object will be instantiated and called as such:* KthLargest* obj = new KthLargest(k, nums);* int param_1 = obj->add(val);*/

创建堆的方法

class KthLargest {
public:void BuildMinHeap(vector<int> &input, vector<int> &heap, int k) {for (int i = 0; i < k; i++) {if(i < input.size()) {heap.push_back(input[i]);}else{heap.push_back(INT_MIN);}}sort(heap.begin(), heap.end());//reverse(heap.begin(), heap.end());}void ShiftMinDown(vector<int> &heap) {int cur_root = 0;int cur_left = 2 * cur_root + 1;int cur_right = 2 * cur_root + 2;while (true) {if(cur_left < heap.size() && cur_right < heap.size() && heap[cur_left] <= heap[cur_right]) {if (heap[cur_root] > heap[cur_left]) {int temp = heap[cur_root];heap[cur_root] = heap[cur_left];heap[cur_left] = temp;cur_root = cur_left;cur_left = 2 * cur_root + 1;cur_right = 2 * cur_root + 2;} else{break;}}else if(cur_left < heap.size() && cur_right >= heap.size()){if(heap[cur_root] > heap[cur_left]) {int temp = heap[cur_root];heap[cur_root] = heap[cur_left];heap[cur_left] = temp;cur_root = cur_left;cur_left = 2 * cur_root + 1;cur_right = 2 * cur_root + 2;}break;}else if (cur_right < heap.size() && heap[cur_root] > heap[cur_right]) {int temp = heap[cur_root];heap[cur_root] = heap[cur_right];heap[cur_right] = temp;cur_root = cur_right;cur_left = 2 * cur_root + 1;cur_right = 2 * cur_root + 2;} else {break;}}}KthLargest(int k, vector<int>& nums) {BuildMinHeap(nums,heap,k);for(int i = k; i<nums.size(); i++){if(heap[0] < nums[i]){heap[0] = nums[i];ShiftMinDown(heap);}}}int add(int val) {if(heap[0] < val){heap[0] = val;ShiftMinDown(heap);}cout<<heap[0]<<endl;return heap[0];}
private:vector<int> heap;
};/*** Your KthLargest object will be instantiated and called as such:* KthLargest* obj = new KthLargest(k, nums);* int param_1 = obj->add(val);*/

 

这篇关于leetcode题:703. 数据流中的第K大元素(简单)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/812452

相关文章

基于Qt开发一个简单的OFD阅读器

《基于Qt开发一个简单的OFD阅读器》这篇文章主要为大家详细介绍了如何使用Qt框架开发一个功能强大且性能优异的OFD阅读器,文中的示例代码讲解详细,有需要的小伙伴可以参考一下... 目录摘要引言一、OFD文件格式解析二、文档结构解析三、页面渲染四、用户交互五、性能优化六、示例代码七、未来发展方向八、结论摘要

MyBatis框架实现一个简单的数据查询操作

《MyBatis框架实现一个简单的数据查询操作》本文介绍了MyBatis框架下进行数据查询操作的详细步骤,括创建实体类、编写SQL标签、配置Mapper、开启驼峰命名映射以及执行SQL语句等,感兴趣的... 基于在前面几章我们已经学习了对MyBATis进行环境配置,并利用SqlSessionFactory核

哈希leetcode-1

目录 1前言 2.例题  2.1两数之和 2.2判断是否互为字符重排 2.3存在重复元素1 2.4存在重复元素2 2.5字母异位词分组 1前言 哈希表主要是适合于快速查找某个元素(O(1)) 当我们要频繁的查找某个元素,第一哈希表O(1),第二,二分O(log n) 一般可以分为语言自带的容器哈希和用数组模拟的简易哈希。 最简单的比如数组模拟字符存储,只要开26个c

csu 1446 Problem J Modified LCS (扩展欧几里得算法的简单应用)

这是一道扩展欧几里得算法的简单应用题,这题是在湖南多校训练赛中队友ac的一道题,在比赛之后请教了队友,然后自己把它a掉 这也是自己独自做扩展欧几里得算法的题目 题意:把题意转变下就变成了:求d1*x - d2*y = f2 - f1的解,很明显用exgcd来解 下面介绍一下exgcd的一些知识点:求ax + by = c的解 一、首先求ax + by = gcd(a,b)的解 这个

hdu2289(简单二分)

虽说是简单二分,但是我还是wa死了  题意:已知圆台的体积,求高度 首先要知道圆台体积怎么求:设上下底的半径分别为r1,r2,高为h,V = PI*(r1*r1+r1*r2+r2*r2)*h/3 然后以h进行二分 代码如下: #include<iostream>#include<algorithm>#include<cstring>#include<stack>#includ

usaco 1.3 Prime Cryptarithm(简单哈希表暴搜剪枝)

思路: 1. 用一个 hash[ ] 数组存放输入的数字,令 hash[ tmp ]=1 。 2. 一个自定义函数 check( ) ,检查各位是否为输入的数字。 3. 暴搜。第一行数从 100到999,第二行数从 10到99。 4. 剪枝。 代码: /*ID: who jayLANG: C++TASK: crypt1*/#include<stdio.h>bool h

uva 10387 Billiard(简单几何)

题意是一个球从矩形的中点出发,告诉你小球与矩形两条边的碰撞次数与小球回到原点的时间,求小球出发时的角度和小球的速度。 简单的几何问题,小球每与竖边碰撞一次,向右扩展一个相同的矩形;每与横边碰撞一次,向上扩展一个相同的矩形。 可以发现,扩展矩形的路径和在当前矩形中的每一段路径相同,当小球回到出发点时,一条直线的路径刚好经过最后一个扩展矩形的中心点。 最后扩展的路径和横边竖边恰好组成一个直

poj 1113 凸包+简单几何计算

题意: 给N个平面上的点,现在要在离点外L米处建城墙,使得城墙把所有点都包含进去且城墙的长度最短。 解析: 韬哥出的某次训练赛上A出的第一道计算几何,算是大水题吧。 用convexhull算法把凸包求出来,然后加加减减就A了。 计算见下图: 好久没玩画图了啊好开心。 代码: #include <iostream>#include <cstdio>#inclu

uva 10130 简单背包

题意: 背包和 代码: #include <iostream>#include <cstdio>#include <cstdlib>#include <algorithm>#include <cstring>#include <cmath>#include <stack>#include <vector>#include <queue>#include <map>

leetcode-24Swap Nodes in Pairs

带头结点。 /*** Definition for singly-linked list.* public class ListNode {* int val;* ListNode next;* ListNode(int x) { val = x; }* }*/public class Solution {public ListNode swapPairs(L