Today~ is a good day!
蒟蒻笔记再度更新之~c++ 优先队列(priority_queue)
优先队列跟普通的队列唯一的差别呢实际上就在于,他能够根据所给的某种类型的优先级规则在访问之前进行一定的排序工作。
具体用途的话:
升序队列
priority_queue <int,vector < int > ,greater < int > > q;//小顶堆 1 2 3 4 5
降序队列
priority_queue <int,vector < int >,less < int > >q;//大顶堆 5 4 3 2 1
大顶堆 是优先队列默认的 可写为
priority_queue q;//大顶堆
- 合并果实
对问题思路简单讲一下吧,在多堆果实中每次挑选两堆最少的进行合并,再将合并完成的小堆push到优先队列里,一直往复。实际上就是一个堆的问题,优先队列也是相当于堆的。本蒟蒻开始想的时候是,用排序写,但是题目时间实在是卡的死死的,没有半点办法~(废话不多上代码)
#include<bits/stdc++.h>
using namespace std;
int n,x,ans;
int main() {
ans=0;
cin>>n;
priority_queue<int,vector<int>,greater<int> >q;//小顶堆
for(int i=1; i<=n; i++) cin>>x,q.push(x);
while(q.size()>=2) {
int a=q.top();
q.pop();
int b=q.top();
q.pop();
ans+=a+b;
q.push(a+b);
}
cout<<ans<<endl;
return 0;
}
- 哈夫曼树(原题没找到暂不更新)
讲下思路就是:所有数据存入优先队列以后,去除队头俩个元素a,b,a+b作为a,b的父节点,并push到优先队列,a,b分别为左右儿子。