priority_queue的模拟实现

最新推荐文章于 2024-08-06 16:24:59 发布

c无序

最新推荐文章于 2024-08-06 16:24:59 发布

阅读量469

点赞数 11

分类专栏： C++ 文章标签： c++

本文链接：https://blog.csdn.net/2202_75331338/article/details/139500943

版权

C++ 专栏收录该内容

34 篇文章 1 订阅

订阅专栏

priority_queue类定义

priority_queue构造函数

priority_queue类push()函数

priority_queue类pop()函数

priority_queue类size()函数

priority_queue类empty()函数

priority_queue类top()函数

仿函数与priority_queue类模拟实现

项目代码：

实现priority_queue的方式和stack和queue基本类似，因为priority_queue也是一种容器适配器，但是只能使用vector和deque，这里默认使用vector容器作为适配器，因为priority_queue本质是堆结构，所以实现时主要按照堆的实现思路进行

priority_queue类定义

//vector作为默认的容器适配器
template<class T, class Container = vector<T>>
class priority_queue
{
private:
    Container _con:
}

priority_queue构造函数

priority_queue主要有两种构造函数

1.无参构造函数

2.迭代器构造函数

需要注意的是，如果不显式实现两个构造函数，则默认是无参构造，该默认构造将会调用容器适配器的构造函数

    //有迭代器构造就必须写显式写空参构造
    priority_queue()
    :_con()
    {
        
    }
    
    //迭代器构造函数
    template<class Iterator>
    priority_queue(Iterator begin, Iterator end)
    :_con(begin, end)
    {
        //向下调整算法建堆
        for (int i = (size()-2)/2; i >=0 ; i--)
        {
            adjust_down(i);
        }
    }

priority_queue类push()函数

    void adjust_up(int child){
        size_t parent =  (child - 1)/2;
        
        while (child > 0) {
            if (_con[parent] <= _con[child]) {
                swap(_con[parent], _con[child]);
                child = parent;
                parent = (child-1)/2;
            }
            else{
                break;
            }
        }
    }
    
    void push(const T& x){
        _con.push_back(x);
        //向上调整建堆
        adjust_up(_con.size() - 1);
    }

priority_queue类pop()函数

与堆删除一样，先交换根节点与最后一个节点数据，在删除最后一个数据，向下调整建堆

void adjust_down(int parent){
        size_t child = parent * 2 + 1;
        while(child < _con.size()){
            if (child+1 < _con.size() && _con[child] < _con[child+1] ){
                ++child;
            }
            
            if(_con[parent] < _con[child]){
                swap(_con[parent], _con[child]);
                parent = child;
                child = parent * 2 + 1;
            }else{
                break;
            }
        }
    }
    void pop(){
        swap(_con[0], _con[_con.size()-1]);
        _con.pop_back();
        adjust_down(0);
    }

priority_queue类size()函数

//size()函数
const size_t size()
{
    return _con.size();
}

priority_queue类empty()函数

//empty()函数
bool empty()
{
    return _con.empty();
}

priority_queue类top()函数

//top()函数_const版本
const T& top() const
{
    return _con[0];
}
 
//top()函数_非const版本
T& top()
{
    return _con[0];
}

仿函数与priority_queue类模拟实现

以前想控制大堆小堆，只能手动修改代码中的符号，有没有一种方法可以让我们更快更高效的控制大于小于呢？当然可以这时候我们就可以使用仿函数，也叫做函数对象，重载了operator( )的类，类的对象可以像函数一样使用

operator()特点，参数个数和返回值根据需求确定，不固定，很灵活

看个例子

class Func{
public:
void operator()(){
    cout << "Fun调用" << endl;
}
};

int main() {
    Func f1;
    f1();//类的对象可以像函数一样使用
    return 0;
}
输出结果：
Fun调用

例如对于比较两个数值，小于返回true的仿函数

template <class T>
class myless{
public:
    
    bool operator()(const T& x,const T& y){
        cout << "operator() < 调用" << endl;
        return x < y;
    }
};

同理返回一个大于true的仿函数

template <class T>
class mygreater{
public:
    bool operator()(const T& x,const T& y){
        cout << "operator() > 调用" << endl;
        return x > y;
    }
};

int main() {
    myless<int> mylessFun;
    cout << mylessFun(2,5) << endl;
    mygreater<int> mygreaterFun;
    cout << mygreaterFun(2,5) << endl;
    return 0;
}
输出结果：
operator() < 调用
1
operator() > 调用
0

void Test2(){
    vector<int> v1 = {34,5,2,41,32,43,21,0};
    nanyi::priority_queue<int,vector<int>,nanyi::myless<int>> pq1;
    for(auto& e:v1){
        pq1.push(e);
    };
    while (!pq1.empty()) {
        cout << pq1.top() << " ";
        pq1.pop();
    }
    cout << endl;
    
}

int main() {
    Test2();
    return 0;
}

输出结果：
43 41 34 32 21 5 2 0

往仿函数传自定义类型Date*

class Date
{
public:
    Date(int year = 1900, int month = 1, int day = 1)
        : _year(year)
        , _month(month)
        , _day(day)
    {}

    bool operator<(const Date& d)const
    {
        return (_year < d._year) ||
            (_year == d._year && _month < d._month) ||
            (_year == d._year && _month == d._month && _day < d._day);
    }

    bool operator>(const Date& d)const
    {
        return (_year > d._year) ||
            (_year == d._year && _month > d._month) ||
            (_year == d._year && _month == d._month && _day > d._day);
    }

    friend ostream& operator<<(ostream& _cout, const Date& d);
private:
    int _year;
    int _month;
    int _day;
};

ostream& operator<<(ostream& _cout, const Date& d)
{
    _cout << d._year << "-" << d._month << "-" << d._day;
    return _cout;
}

struct PDateLess
{
    bool operator()(Date* p1, Date* p2)
    {
        return *p1 < *p2;
    }
};

void TestPriorityQueue()
{
    // 大堆，需要用户在自定义类型中提供<的重载
    nanyi::priority_queue<Date*, vector<Date*>, PDateLess> q1;
    q1.push(new Date(2018, 10, 29));
    q1.push(new Date(2018, 10, 28));
    q1.push(new Date(2018, 10, 30));

    while (!q1.empty())
    {
        cout << *q1.top() << " ";
        q1.pop();
    }
    cout << endl;
}

int main()
{
    TestPriorityQueue();
    
    return 0;
}

项目代码：

//
//  priority_queue.hpp
//  优先级队列
//
//  Created by 南毅 on 2024/6/4.
//

#ifndef priority_queue_hpp
#define priority_queue_hpp

#include <iostream>
using namespace std;

#endif /* priority_queue_hpp */

namespace nanyi{

template <class T>
class myless{
public:
    
    bool operator()(const T& x,const T& y){
        //cout << "operator() < 调用" << endl;
        return x < y;
    }
};

template <class T>
class mygreater{
public:
    bool operator()(const T& x,const T& y){
        //cout << "operator() > 调用" << endl;
        return x > y;
    }
};

template<class T,class Container = vector<T>,class Compare = myless<T>>

class priority_queue{
private:
    Container _con;
public:
    
    void adjust_up(int child){
        Compare confunc;
        size_t parent =  (child - 1)/2;
        
        while (child > 0) {
           // if (_con[parent] <= _con[child])
            if (confunc(_con[parent] , _con[child]))
            {
                swap(_con[parent], _con[child]);
                child = parent;
                parent = (child-1)/2;
            }
            else{
                break;
            }
        }
    }
    
    void push(const T& x){
        _con.push_back(x);
        //向上调整建堆
        adjust_up(_con.size() - 1);
    }
    
    void adjust_down(int parent){
        Compare confunc;
        size_t child = parent * 2 + 1;
        while(child < _con.size()){
            if (child+1 < _con.size() && confunc(_con[child] , _con[child+1] )){
                ++child;
            }
            
            //if(_con[parent] < _con[child])
            if(confunc(_con[parent] , _con[child]))
            {
                swap(_con[parent], _con[child]);
                parent = child;
                child = parent * 2 + 1;
            }else{
                break;
            }
        }
    }
    void pop(){
        swap(_con[0], _con[_con.size()-1]);
        _con.pop_back();
        adjust_down(0);
    }
    
    const T& top(){
        return _con[0];
    }
    
    size_t size(){
        return _con.size();
    }
    
    bool empty(){
        return _con.empty();
    }
    
    //有迭代器构造就必须写显式写空参构造
    priority_queue()
    :_con()
    {
        
    }
    
    //迭代器构造函数
    template<class Iterator>
    priority_queue(Iterator begin, Iterator end)
    :_con(begin, end)
    {
        //向下调整算法建堆
        for (int i = (size()-2)/2; i >=0 ; i--)
        {
            adjust_down(i);
        }
    }
    
    
};
}

c无序

关注

11
点赞
踩
7

收藏

觉得还不错? 一键收藏
0
评论
priority_queue的模拟实现

目录priority_queue类定义priority_queue构造函数priority_queue类push()函数priority_queue类pop()函数priority_queue类size()函数priority_queue类empty()函数priority_queue类top()函数仿函数与priority_queue类模拟实现项目代码：实现priority_queue的方式和stack和queue基本类似，因为priority_queue也是一种容器适配器，但是只能使用vector和de
复制链接

扫一扫

专栏目录