哈夫曼编码的C++实现(优先队列、贪心)

哈夫曼编码C++实现

哈夫曼编码就是不断地取数列中最小的两个,合并,再添加到数列中,循环。利用了贪心的思想。在C++实现的时候可以用到优先队列 priority_queue,这个方便的东西。

这个优先队列,会自己进行排序,每次加进去一个数,都会更新,降序排列或升序排列。以升序排列为例,更新过后的队首元素就是数列中最小的数。

priority_queue的用法

首先要用到头文件#include<queue>
定义优先队列:priority_queue<int,vector<int>,greater<int> >q;(这里不需要用到<vector>头文件)其中,greater<int>是升序(从小到大排序),如果是降序(从大到小)就要用less<int>

要注意的是最后两个尖括号不要挨着> >,因为>>是重载运算符,编译器会报错。

然后是一些优先队列的基本操作:

	priority_queue<int,vector<int>,greater<int> >q;//定义优先队列
	q.push(x);//将x加入到优先队列中
	t=q.top();//返回队首元素
	q.pop();//弹出队首元素
	l=q.size();//返回队列长度(元素个数)
	b=q.empty();//判断队列是否为空,为空返回1,否则返回0

然后就可以上代码了:

#include <queue>
#include <iostream>
using namespace std;
int main()
{
    
    
    int n,a,b,c;
    priority_queue<int,vector<int>,greater<int> >q;
    cin>>n;
    for(int i=0;i<n;i++){
    
    
        cin>>c;
        q.push(c);
    }
    int ans=0;
    while(q.size()>1){
    
    
        a=q.top();q.pop();//取出最小的两个元素
        b=q.top();q.pop();
        ans=a+b;//合并
        q.push(ans);//将合并后的数加入到队列中
    }
    cout<<ans<<endl;//最后编码总长度
    return 0;
}

代码很短,很简单。

猜你喜欢

转载自blog.csdn.net/weixin_45743427/article/details/108970237
今日推荐