在构造Huffman树的时候,需要把每个根节点(指针,也不知道叫啥了)存到priority_queue里面,那么想要按照每棵树根节点的data值从小到大排,这样才能够每次直接从队首取出当前根的data最小的两棵树,那么priority_queue需要做处理
//首先这个是每个结点的构造吧
typedef struct node
{
int data;//data域
struct node *lchild,*rchild;//左孩子,右孩子指针
}Node;
然后自己构造比较函数:
struct Compare{
bool operator()(Node* &a,Node* &b)
{
return a->data > b->data;//让他们按照data的值从小到大排
}
};
然后就可以用了:
priority_queue<Node*,vector<Node*>,Compare> mq;
//创建了一个优先队列mq,里面存的是指针,也就是接下来要存放当前的树的跟指针,按照根节点的data自动从小到大排列
最后是pta里面的一道题:
构造哈夫曼树,然后输出它树的中序序列。
从小到大的顺序给出词频(不超过10个),根据词频构造哈夫曼树。
为确保构建的哈夫曼树唯一,本题做如下限定:
(1)选择根结点权值最小的两棵二叉树时,选取权值较小者作为左子树。
(2)若多棵二叉树根结点权值相等,按先后次序分左右,先出现的作为左子树,后出现的作为右子树。
输入格式:
第一行输入词频个数;
第二行按从小到大的顺序输入每个词频
输出格式:
输出中序序列,中间以一个空格隔开
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
typedef struct node
{
int data;
struct node *lchild,*rchild;
}Node,*Tree;
struct Compare{
bool operator()(Node* &a,Node* &b)
{
return a->data > b->data;
}
};
//我想要按照每一棵树的根节点值域的大小把他们排起来,优先队列,但是我该怎么搞小根堆呢
//不确定他是否会按照data去从小到大排
Tree merge(Tree a,Tree b)
{
Node *C=(Tree)malloc(sizeof(Node));
int aa=a->data;
int bb=b->data;
int cc=aa+bb;
C->data=cc;
C->lchild=a;
C->rchild=b;
return C;
}
void creat_tree(priority_queue<Node*,vector<Node*>,Compare> &mq)
{
Tree A=mq.top();mq.pop();
Tree B=mq.top();mq.pop();
Tree C=merge(A,B);
mq.push(C);
}
void zfound(Tree &T)
{
if(T)
{
zfound(T->lchild);
cout<<T->data<<' ';
zfound(T->rchild);
}
}
int main()
{
int n;
cin>>n;
priority_queue<Node*,vector<Node*>,Compare> mq;
for(int i=1;i<=n;i++)
{
int x;
cin>>x;
Node *a=(Node*)malloc(sizeof(Node));
a->data=x;
a->lchild=a->rchild=NULL;
mq.push(a);//把这个结点的指针吧算是,放到队列里面
}
while(mq.size()>1)
{
creat_tree(mq);
}
Tree T;//是最终的那一棵树,就是哈夫曼树
T=mq.top();
zfound(T);
return 0;
}