标签:2.0 top 中位数 large small 295 Leetcode size
题目:
思路:
用一个大根堆装较小的数,用小根堆装较大的数,维持两个堆的大小接近。
那么取中位数时:
- large.size==small.size,(large.top+small.top)/2.0 注意2.0为了获取double
- large>small,返回large.top
- large<small,返回small.top
每次addNum时,如果small.size<large.size,那么先将large.push(num),再n=large.top,large.pop,保证取出的值是较大数中最小值,将small.push(n)。
反之亦然
class MedianFinder { public: MedianFinder() { } void addNum(int num) { if(large.size()<small.size()){ small.push(num); int n=small.top(); small.pop(); large.push(n); }else{ large.push(num); int n=large.top(); large.pop(); small.push(n); } } double findMedian() { // 如果元素不一样多,多的那个堆的堆顶元素就是中位数 if(large.size()<small.size()){ return small.top(); }else if(large.size()>small.size()){ return large.top(); }else{ return (small.top()+large.top())/2.0; } } // 小顶堆 priority_queue<int> large; // 大顶堆 priority_queue<int,vector<int>,greater<int>> small; };
标签:2.0,top,中位数,large,small,295,Leetcode,size 来源: https://www.cnblogs.com/zl1991/p/15933222.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。