find-median-from-data-stream & multiset priority queue 堆

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;
}