【模板】【数据结构】堆

最新推荐文章于 2024-09-16 18:31:59 发布

HFYRKG

最新推荐文章于 2024-09-16 18:31:59 发布

阅读量225

点赞数

分类专栏：模板文章标签：数据结构模板

本文链接：https://blog.csdn.net/george__yu/article/details/77179526

版权

模板专栏收录该内容

8 篇文章 0 订阅

订阅专栏

1.手写小根堆：

#include <cstdio>
#include <algorithm>
#define maxn 1000003
using namespace std;
int heap[maxn];
int tot=0;
inline void pushup(int x)
{
    for(;x>0;x>>=1) 
    {
        if(heap[x]>heap[x>>1]) break;
        swap(heap[x],heap[x>>1]);
    }
}
inline void pushdown(int x)
{
    for(int y;x<=tot;)
    {
        y=x<<1;
        if(y>tot) break;
        if(heap[y|1]<heap[y]&&y<tot) y|=1;
        if(heap[x]<heap[y]) break;
        swap(heap[x],heap[y]);
        x=y;
    }
}
inline void push(int x)
{
    heap[++tot]=x;
    pushup(tot);
}
inline int top()
{
    return heap[1];
}
inline void pop()
{
    swap(heap[1],heap[tot--]);
    pushdown(1);
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        int opt,x;
        scanf("%d",&opt);
        switch(opt)
        {
            case 1: {scanf("%d",&x);push(x);break;}
            case 2: {printf("%d\n",top());break;}
            case 3: {pop();break;}
            default:break;
        }
    }
    return 0;
}

2.STL的priority_queue

#include <queue>
using namespace std;
struct node{
    int a,b,c;
    bool operator < (const node &x) const{
        return a.x<x;
    }
};
priority_queue<int>q1;//定义一个整数类型的大根堆
priority_queue<int,vector<int>,greater<int> >q2//定义一个整数类型的小根堆，注意两个尖括号中间一定要加空格，不然容易CE
priority_queue<node>q3;//定义一个node类型的优先队列，node必须重载过小于号。这里的q3是个小根堆