priority_queue 用法与编程题目

推荐阅读:C++ priority_queue 与 lambda的结合使用

#include <queue> 头文件

priority_queue<Type, Container, Functional>

默认是大顶堆

greater<>成为小顶堆priority_queue<int, vector<int>, greater<int> > q;

加入 q.push();

弹出 q.pop();

队头 q.top();

对于自定义类型,则必须自己重载 operator< 或者自己写仿函数

这里得仿函数做的是判断当前值得大小往后放,因此,greater<>,大的元素往后放,就是小顶堆。

struct Node{
    int x, y;
    Node(int a= 0, int b= 0 ):
        x(a), y(b) {}
};
 
bool operator<( Node a, Node b ){
    if( a.x== b.x ) return a.y> b.y;//小顶堆
    return a.x> b.x; 
}


也可以使用函数对象实现:

#include <iostream>
#include <queue>
 
using namespace std;
 
struct Node{
    int x, y;
    Node( int a= 0, int b= 0 ):
        x(a), y(b) {}
};
 
struct cmp{
    bool operator() ( Node a, Node b ){
        if( a.x== b.x ) return a.y> b.y;
         
        return a.x> b.x; }// 小顶堆
};
 
int main(){
    priority_queue<Node, vector<Node>, cmp> q;
     
    for( int i= 0; i< 10; ++i )
    q.push( Node( rand(), rand() ) );
     
    while( !q.empty() ){
        cout << q.top().x << ' ' << q.top().y << endl;
        q.pop();
    }
     
    getchar();
    return 0;
} 

例题:

1、美团2016  字符编码

请设计一个算法,给一个字符串进行二进制编码,使得编码后字符串的长度最短。


输入描述:
每组数据一行,为待编码的字符串。保证字符串长度小于等于1000。




输出描述:
一行输出最短的编码后长度。


输入例子1:
MT-TECH-TEAM


输出例子1:
33

自己的思路:

自己独立更改了很多次,核心的方法是哈夫曼编码。首先使用一个map统计所有字符出现的次数,放入一个小顶堆,堆的值按照次数排序。每次弹出两个最小的两个,构建成一个节点后再加入堆中。当堆为空时,说明所有节点都处理完成。

更改很多次的原因是,怎么迎合题目输出的要求。举例:一个节点的次数是2,当他和其他节点构成一个新的节点时,编码长度的改变是:这个节点对应的两个相同字符都加一。

因此,堆中存储的数据更改为一个结构体{sum,node},sum表示这个节点目前的编码长度统计(默认带有前支),node代表当前节点的权值;

每次构建一个新节点时,新节点的sum=两个节点的sum之和,同时要加上两个节点的权值之和(因为有产生一个向上的分支,这个权值代表的是当前这个综合节点的字符数量),新的节点探入{新的sum,权值之和}

特殊情况:当队列为空时,只有一个节点了,这个时候的sum只是两个sum的和,不能再加上两个节点的权值和(根节点没有向上的分支)

代码如下:

#include<iostream>
#include<vector>
#include<string>
#include<queue>
#include<map>
using namespace std;
struct code{
int sum;
int node;

};
bool operator<(code a,code b){
    if(a.node==b.node) return a.sum>b.sum;
    return a.node>b.node;
}
int main()
{
//哈夫曼编码
    string input;
    while(cin>>input)
    {
        map<char,int> sta;
    priority_queue<code,vector<code>> q;
    for(auto s:input)
    {
        sta[s]++;
    }
    for(auto s:sta)
        q.push({s.second,s.second});
    int n=q.size()-1;
    int sum=0;
    if(n==0) sum=1;
    while(n!=0&&!q.empty())
    {
        code left=q.top();
        q.pop();
        code right=q.top();
        q.pop();
        int temp=left.sum+right.sum+left.node+right.node;
        if(!q.empty())
            q.push({temp,left.node+right.node});
        if(q.empty())
            sum=temp-left.node-right.node;//到根节点,将向上分支多加的减掉
    }
    cout<<sum<<endl;
    }

    //system("pause");
    return 0;
}

378. Kth Smallest Element in a Sorted Matrix

http://blog.csdn.net/m0_37693059/article/details/77772048

使用了

priority_queue<pair<int, pair<int, int>>> myQueue;

make_pair ,emplace,lambda表达式等

greater<pair<int,int>> 默认比较第一个值


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值