堆Heap

#pragma once
#include <vector>

// 小堆
template<class T>  //仿函数
struct Less
{
       bool operator() (const T& l, const T& r)
       {
             return l < r; // operator<
       }
};

template<class T>
struct Greater
{
       bool operator() (const T& l, const T& r)
      {
           return l > r; // operator<
      }
};

template<class T, class Compare = Less<T>>
class Heap
{
public:
       Heap()
       {}

       Heap(const T* a, size_t size)
      {
             for (size_t i = 0; i < size; ++i)
             {
                  _arrays.push_back(a[i]);
             }

              // 建堆
             for(int i = (_arrays.size()-2)/2; i >= 0; --i)
            {
                  AdjustDown(i);
            }
      }

void Push(const T& x)
{
       _arrays.push_back(x);
      AdjustUp(_arrays.size()-1);
}

void Pop()
{
      assert(_arrays.size() > 0);
      swap(_arrays[0], _arrays[_arrays.size() - 1]);
      _arrays.pop_back();

      AdjustDown(0);
}

T& Top()
{
       assert(_arrays.size() > 0);
       return _arrays[0];
}

bool Empty()
{
       return _arrays.empty();
}

size_t Size()
{
      return _arrays.size();
}

void AdjustDown(int root)
{
      int child = root*2 + 1;
//
      Compare com;
      while (child < _arrays.size())
     {
            // 比较出左右孩子中小的那个
           //if (child+1<_arrays.size() &&
           // _arrays[child+1] > _arrays[child])
           if(child+1<_arrays.size() &&
           com(_arrays[child+1],_arrays[child]))
          {
                ++child;
          }

          //if(_arrays[child] > _arrays[root])
          if(com(_arrays[child],_arrays[root]))
         {
                swap(_arrays[child], _arrays[root]);
                root = child;
               child = 2*root+1;
         }
         else
        {
              break;
        }
    }
}

void AdjustUp(int child)
{
        int parent = (child-1)/2;

        //while (parent >= 0)
        while (child > 0)
       {
              //if (_arrays[child] > _arrays[parent])
               if (Compare()(_arrays[child],_arrays[parent]))
               {
                      swap(_arrays[parent], _arrays[child]);
                      child = parent;
                      parent = (child-1)/2;
               }
               else
              {
                     break;
              }
       }
}

void Print()
{
       for (size_t i = 0; i < _arrays.size(); ++i)
      {
           cout<<_arrays[i]<<" ";
      }
      cout<<endl;
}

public:
/*T* _array;
size_t _size;
size_t _capacity;*/
vector<T> _arrays;
};

template<class T>
class PriorityQueue    //优先级队列
{
public:
      void Push(const T& x)
     {
           _hp.Push(x);
     }

     void Pop()
    {
          _hp.Pop();
    }

public:
     Heap<T> _hp;
};

void Test1()
{
        int a[10] = {10, 11, 13, 12, 16, 18, 15, 17, 14, 19};
        Heap<int, Greater<int> > hp1(a, 10);
        hp1.Push(1);
        hp1.Print();

        Heap<int> hp2(a, 10);
        hp2.Push(1);
        hp2.Print();

//Less<int> less;
//cout<<less(1, 2)<<endl;

//Greater<int> greater;
//cout<<greater(1, 2)<<endl;
}

#include <list>

void Test2()
{
       vector<int> v1;
       v1.push_back(1);
       v1.push_back(2);
       v1.push_back(3);
       v1.push_back(4);

// [)
      vector<int>::iterator it = v1.begin();//迭代器
      while (it != v1.end())
      {
            cout<<*it<<" ";
            ++it;
      }
     cout<<endl;

      list<int> l1;
      l1.push_back(1);
      l1.push_back(2);
      l1.push_back(3);
      l1.push_back(4);

      list<int>::iterator listIt = l1.begin();
      while (listIt != l1.end())
     {
           cout<<*listIt<<" ";
          ++listIt;
     }

     cout<<endl;
}