码迷,mamicode.com
首页 > 编程语言 > 详细

最大堆/最小堆/优先队列 实现代码(c++)

时间:2015-04-05 17:16:34      阅读:151      评论:0      收藏:0      [点我收藏+]

标签:

自我感觉代码写的比较乱,这方面要好好注意一下。

总结:

1.在使用vector<int>::size_type 类似的类型时,千万要注意循环的条件判断,很容易发生溢出的危险!所以我最后很懒的选择使用int  - -。

2.下标表示和元素个数表示之间的细微差别。下标之间的变换关系:

   父节点 parent(i)=(i-1)/2; 左孩子 left(i)=2*i+1;右孩子 right(i)=2*i+2

class Max_Heap{
    typedef int index;
public:
    Max_Heap(const vector<int> &vec) :v(vec){
        Build_Heapify(v);
    }
    ~Max_Heap(){};
    void Heap_Sort();
    int Heap_Max() { return v[0]; }
    int Heap_Extract_Max();
    void Insert(int key);
private:
    vector<int> v;
    void Max_Heapify(vector<int> &vec, index i, int hs);
    void Build_Heapify(vector<int> &vec);
};

void
Max_Heap::Max_Heapify(vector<int> &vec, index i, int hs){
    int l = 2 * i + 1, r = 2 * i + 2;
    int largest = i;
    if (l < hs&&vec[l] > vec[i])
        largest = l;
    if (r < hs&&vec[r] > vec[largest])
        largest = r;
    if (largest != i){
        swap(vec[i], vec[largest]);
        Max_Heapify(vec, largest, hs);
    }
}

void
Max_Heap::Build_Heapify(vector<int> &vec){
    int hs = vec.size();
    for (int i = (hs / 2) - 1; i >= 0; --i)
        Max_Heapify(vec, i, hs);
}

void
Max_Heap::Heap_Sort(){
    int hs = v.size();
    Build_Heapify(v);
    for (int i = hs - 1; i >= 0; --i){
        swap(v[0], v[i]);
        Max_Heapify(v, 0, i);
    }
}
int
Max_Heap::Heap_Extract_Max() {
    if (v.size() == 0)
    {
        std::cout << "over flow";
        return 0;
    }
    else {
        int temp = v[0];
        swap(v[0], v[v.size() - 1]);
        v.pop_back();
        Max_Heapify(v, 0, v.size());
        return temp;
    }
}
void
Max_Heap::Insert(int key){
    v.push_back(key);
    int sz = v.size();
    while (sz>=2&&key > v[sz / 2 - 1]){
        swap(v[sz - 1], v[sz / 2 - 1]);
        sz = sz / 2;
    }
}

 

最大堆/最小堆/优先队列 实现代码(c++)

标签:

原文地址:http://www.cnblogs.com/Nastukashii/p/4394307.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!