基于最大堆实现最大优先队列

这个没啥好说的，写完上一个正好写一下这一个，都是基于堆实现的，不过这里没有用C++内置数组，原因是添加删除元素有点麻烦，所以用了vector来实现。

详细内容见《算法导论》P90

----------------------------------------------------代码-----------------------------------------------------------

// 基于最大堆实现最大优先队列.cpp: 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <iostream>
#include <vector>

using namespace std;

int PARENT(int i);
int LEFT(int i);
int RIGHT(int i);
int MAX_HEAPIFY(vector<int>& A, int i);

class PriorityQueue  //优先队列  
{
protected:
     vector<int> Queue;
public:
    void push(int x)
    {
        Queue.push_back(x);
        MAX_HEAPIFY(Queue, 0);
        print();
    }
    int pop()
    {
        if (Queue.size() > 0)
        {
            int top = Queue[0];
            Queue[0] = Queue[Queue.size() - 1];
            Queue.pop_back();
            MAX_HEAPIFY(Queue, 0);
            print();
            return top;
        }
        else
        {
            cout << "队列为空!" << endl;
            return 0;
        }
    }
    int getTop()
    {
        if (Queue.size() > 0)
        {
            return Queue[0];
        }
    }
    void set_key(int i,int key)
    {
        Queue[i] = key;
        MAX_HEAPIFY(Queue, 0);
        print();
    }
    void print()
    {
        for (auto c : Queue)
            cout << c << ends;
        cout << endl;
    }
};

//--------------------------------------------------------------

int PARENT(int i)
{
    return (i - 1) / 2;
}

int LEFT(int i)
{
    return 2 * i + 1;
}

int RIGHT(int i)
{
    return 2 * i + 2;
}

//维护最大堆，时间复杂度为O(lgn)
int MAX_HEAPIFY(vector<int>& A, int i)
{
    int l, r, largest;//l为左孩子的下标，r为右孩子的下标，largest为三者中最大数的下标
    int temp;
    l = LEFT(i);
    r = RIGHT(i);
    if (l < A.size() && A[l] > A[i])
        largest = l;
    else
        largest = i;

    if (r < A.size() && A[r] > A[largest])
        largest = r;

    if (largest != i)
    {
        temp = A[i];
        A[i] = A[largest];
        A[largest] = temp;
        MAX_HEAPIFY(A, largest);
    }
    else
    {
        if (l >= A.size() && r >= A.size())//到达了叶节点，停止递归
        {
            return 0;
        }
        MAX_HEAPIFY(A, l);
        MAX_HEAPIFY(A, r);
    }
}


int main()
{
    PriorityQueue p;
    p.push(4);
    p.push(1);
    p.push(3);
    p.push(2);
    p.push(16);
    p.push(9);
    p.push(10);
    p.push(14);
    p.push(8);
    p.push(7);
    cout << endl;
    for (int i = 0; i < 10; i++)
        p.pop();
    return 0;
}

运行结果如下：