1 哈夫曼树的建立
利用线性表存储树结点的数组 提前建立树结点
2 根据遍历序列确定二叉树:
1 已知两种遍历序列 :
通过根节点的分割确定左右子树
2 通过”#”动态表示:
遇”#”说明该子树为空 要换成右子树
右子树为”#” 返回上一级的右子树
代码1如下
//算法5.10 构造赫夫曼树
#include<iostream>
using namespace std;
typedef struct
{
int weight;
int parent,lchild,rchild;
}HTNode,*HuffmanTree;
void Select(HuffmanTree HT,int len,int &s1,int &s2)
{
int i,min1=0x3f3f3f3f,min2=0x3f3f3f3f;//先赋予最大值
for(i=1;i<=len;i++)
{
if(HT[i].weight<min1&&HT[i].parent==0)
{
min1=HT[i].weight;
s1=i;
}
}
int temp=HT[s1].weight;//将原值存放起来,然后先赋予最大值,防止s1被重复选择
HT[s1].weight=0x3f3f3f3f;//实现删除s
for(i=1;i<=len;i++)
{
if(HT[i].weight<min2&&HT[i].parent==0)
{
min2=HT[i].weight;
s2=i;
}
}
HT[s1].weight=temp;//恢复原来的值
}
void CreatHuffmanTree(HuffmanTree &HT,int n) // &HT引用指针 HT直接变成数组名
{
//构造赫夫曼树HT
int m,s1,s2,i;
if(n<=1) return;
m=2*n-1; //哈夫曼树结点性质 :一棵有n个叶子结点的哈夫曼树共有2n-1个结点
HT=new HTNode[m+1]; //0号单元未用,所以需要动态分配m+1个单元,HT[m]表示根结点
for(i=1;i<=m;++i) //将1~m号单元中的双亲、左孩子,右孩子的下标都初始化为0
{ HT[i].parent=0; HT[i].lchild=0; HT[i].rchild=0; }
cout<<"请输入叶子结点的权值:\n";
for(i=1;i<=n;++i) //输入前n个单元中叶子结点的权值
cin>>HT[i].weight;
/*――――――――――初始化工作结束,下面开始创建赫夫曼树――――――――――*/
for(i=n+1;i<=m;++i) //控制最后只有一个最小值的情况
{ //通过n-1次的选择、删除、合并来创建赫夫曼树
Select(HT,i-1,s1,s2);
//在HT[k](1≤k≤i-1)中选择两个其双亲域为0且权值最小的结点,
// 并返回它们在HT中的序号s1和s2
HT[s1].parent=i;
HT[s2].parent=i;
//得到新结点i,从森林中删除s1,s2,将s1和s2的双亲域由0改为i
HT[i].lchild=s1;
HT[i].rchild=s2; //s1,s2分别作为i的左右孩子
HT[i].weight=HT[s1].weight+HT[s2].weight; //i 的权值为左右孩子权值之和
} //for
} // CreatHuffmanTree
void main()
{
HuffmanTree HT;
int n;
cout<<"请输入叶子结点的个数:\n";
cin>>n;
CreatHuffmanTree(HT,n);
cout<<"哈夫曼树建立完毕!\n";
return 0;
}
代码2如下
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
unordered_map<int,int> pos;
vector<int> preorder,inorder;
TreeNode* build(int a,int b,int x,int y)// a-b 表示根据前序遍历确定的树(左右子树) x-y同理
{
//每次区间都要缩小 a>b 说明该树没有结点了
if(a>b) return NULL;
auto root =new TreeNode(preorder[a]);
//构造二叉树 内存中没有 和遍历一样每次都能确定根节点 所以每次建子树的根节点
int k=pos[preorder[a]];
root->left=build(a+1,a+1+k-1-x,x,k-1);
root->right=build(a+1+k-1-x+1,b,k+1,y);
return root;
}
TreeNode* buildTree(vector<int>& _preorder, vector<int>& _inorder) {
preorder=_preorder,inorder=_inorder;
int n=inorder.size();
for(int i=0;i<inorder.size();i++) pos[inorder[i]]=i; //逆着存储 记录在数组中的哪个位置
return build(0,n-1,0,n-1);
}
};