https://leetcode.com/problems/find-median-from-data-stream/
这道题目实在是不错,所以单独拎出来。
https://discuss.leetcode.com/topic/27521/short-simple-java-c-python-o-log-n-o-1/2
看这里面的C++解法,用了priority_queue来实现的。(其实也是堆,之前我一直用multiset)
而Java用的PriorityQueue来做。
关于multiset 和 priority queue的区别,可以看这里:
https://discuss.leetcode.com/topic/27521/short-simple-java-c-python-o-log-n-o-1/2
priority_queue 调用 STL里面的 make_heap(), pop_heap(), push_heap() 算法实现。
注意,priority queue默认是大顶堆,top()和pop()都是针对最大的元素。
而Java的PriorityQueue是从小到大排的。
注意下面对于比较方法的初始化(尤其是比较方式参数的传入)
// constructing priority queues #include <iostream> // std::cout #include <queue> // std::priority_queue #include <vector> // std::vector #include <functional> // std::greater class mycomparison { bool reverse; public: mycomparison(const bool& revparam=false) {reverse=revparam;} bool operator() (const int& lhs, const int&rhs) const { if (reverse) return (lhs>rhs); else return (lhs<rhs); } }; int main () { int myints[]= {10,60,50,20}; std::priority_queue<int> first; std::priority_queue<int> second (myints,myints+4); std::priority_queue<int, std::vector<int>, std::greater<int> > third (myints,myints+4); // using mycomparison: typedef std::priority_queue<int,std::vector<int>,mycomparison> mypq_type; mypq_type fourth; // less-than comparison mypq_type fifth (mycomparison(true)); // greater-than comparison return 0; }