数据结构——树和二叉树的操作实现
- 树和二叉树这一章的知识点相较于前几章而言有一点难度,这一章的算法都是以递归的形式呈现的,对算法的理解要求较高。这一章的操作实现以二叉树的操作为主。
一、二叉树的结构定义
树的存储结构可以是顺序存储,也可以链式存储。这一章的算法以链式的为主,所以在这里给出二叉树的链式存储结构定义。这里的TElemType用户需要自己定义,以下的算法中这里的类型均为字符char型。
typedef struct BiTNode
{
TElemType data;
struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;
二、二叉树的先序建立
为了让用户能更好的认识二叉树的先序建立算法,这里描述了这个二叉树的建立过程,即每一步的输入都能让用户看到当前输入的是哪个节点的哪个孩子,有利于这棵树的建立,以防数据多了以后用户不知道这棵二叉树何时停止。
void CreateBiTree(BiTree &T)
{
char ch;
cin>>ch;
if (ch=='#')
T=NULL;
else
{
T=new BiTNode;
T->data=ch;
cout<<"请输入"<<ch<<"的左孩子:";
CreateBiTree(T->lchild);
cout<<"请输入"<<ch<<"的右孩子:";
CreateBiTree(T->rchild);
}
}
三、二叉树的三种遍历
二叉树的遍历相对而言比较简单,三种遍历的过程极其相似,在这里一并呈现。在学习这一章的内容时,大家需要掌握给定一棵树可以写出其三种遍历序列即可,代码只是辅助工具。
void PreOrderTraverse(BiTree T)
{
if (T)
{
cout<<T->data;
PreOrderTraverse(T->lchild);
PreOrderTraverse(T->rchild);
}
}
void InOrderTraverse(BiTree T)
{
if (T)
{
InOrderTraverse(T->lchild);
cout<<T->data;
InOrderTraverse(T->rchild);
}
}
void PostOrderTraverse(BiTree T)
{
if (T)
{
PostOrderTraverse(T->lchild);
PostOrderTraverse(T->rchild);
cout<<T->data;
}
}
四、二叉树的其他基本操作
1. 复制二叉树
void Copy(BiTree T,BiTree &NewT)
{
if (T==NULL)
{
NewT=NULL;
return;
}
else
{
NewT=new BiTNode;
NewT->data=T->data;
Copy(T->lchild,NewT->lchild);
Copy(T->rchild,NewT->rchild);
}
}
2. 求二叉树的深度
int Depth(BiTree T)
{
if (T==NULL)
return 0;
else
{
int m=Depth(T->lchild);
int n=Depth(T->rchild);
return m>n?m+1:n+1;
}
}
3. 求二叉树的结点总数
int NodeCount(BiTree T)
{
if (T==NULL)
return 0;
else
return NodeCount(T->lchild)+NodeCount(T->rchild)+1;
}
4. 求二叉树的叶子结点总数
int LeafCount(BiTree T)
{
if (T==NULL)
return 0;
if (T->lchild==NULL&&T->rchild==NULL)
return 1;
else
return LeafCount(T->lchild)+LeafCount(T->rchild);
}
二叉树的内容还有它的5个基本性质,在这里就不一一阐述了。总之这些性质在做题的时候很常用,大家需要重点掌握这些基础的知识,同时要理清相关的概念,以免混淆。
5. 完整代码
#include <bits/stdc++.h>
#define TElemType char
using namespace std;
typedef struct BiTNode
{
TElemType data;
struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;
void Menu()
{
cout<<"二叉树的操作实现"<<endl;
cout<<"1.先序顺序建立二叉链表"<<endl;
cout<<"2.三种方法遍历二叉树(先序、中序、后序)"<<endl;
cout<<"3.复制二叉树"<<endl;
cout<<"4.计算二叉树的深度"<<endl;
cout<<"5.统计二叉树中结点个数"<<endl;
cout<<"6.统计二叉树中叶子结点个数"<<endl;
cout<<"0.退出"<<endl;
cout<<"请选择服务:";
}
void CreateBiTree(BiTree &T)
{
char ch;
cin>>ch;
if (ch=='#')
T=NULL;
else
{
T=new BiTNode;
T->data=ch;
cout<<"请输入"<<ch<<"的左孩子:";
CreateBiTree(T->lchild);
cout<<"请输入"<<ch<<"的右孩子:";
CreateBiTree(T->rchild);
}
}
void PreOrderTraverse(BiTree T)
{
if (T)
{
cout<<T->data;
PreOrderTraverse(T->lchild);
PreOrderTraverse(T->rchild);
}
}
void InOrderTraverse(BiTree T)
{
if (T)
{
InOrderTraverse(T->lchild);
cout<<T->data;
InOrderTraverse(T->rchild);
}
}
void PostOrderTraverse(BiTree T)
{
if (T)
{
PostOrderTraverse(T->lchild);
PostOrderTraverse(T->rchild);
cout<<T->data;
}
}
void Copy(BiTree T,BiTree &NewT)
{
if (T==NULL)
{
NewT=NULL;
return;
}
else
{
NewT=new BiTNode;
NewT->data=T->data;
Copy(T->lchild,NewT->lchild);
Copy(T->rchild,NewT->rchild);
}
}
int Depth(BiTree T)
{
if (T==NULL)
return 0;
else
{
int m=Depth(T->lchild);
int n=Depth(T->rchild);
return m>n?m+1:n+1;
}
}
int NodeCount(BiTree T)
{
if (T==NULL)
return 0;
else
return NodeCount(T->lchild)+NodeCount(T->rchild)+1;
}
int LeafCount(BiTree T)
{
if (T==NULL)
return 0;
if (T->lchild==NULL&&T->rchild==NULL)
return 1;
else
return LeafCount(T->lchild)+LeafCount(T->rchild);
}
int main()
{
while (true)
{
system("cls");
Menu();
BiTree T;
int x;
cin>>x;
switch(x)
{
case 1:
cout<<"\n请输入根结点:";
CreateBiTree(T);
cout<<"\n创建二叉树成功!\n"<<endl;
break;
case 2:
cout<<"\n二叉树先序遍历序列为:";
PreOrderTraverse(T);
cout<<endl;
cout<<"\n二叉树中序遍历序列为:";
InOrderTraverse(T);
cout<<endl;
cout<<"\n二叉树后序遍历序列为:";
PostOrderTraverse(T);
cout<<"\n"<<endl;
break;
case 3:
BiTree NewT;
Copy(T,NewT);
cout<<"\n原二叉树先序遍历序列为:";
PreOrderTraverse(T);
cout<<endl;
cout<<"新二叉树先序遍历序列为:";
PreOrderTraverse(NewT);
cout<<endl;
cout<<"\n原二叉树中序遍历序列为:";
InOrderTraverse(T);
cout<<endl;
cout<<"新二叉树中序遍历序列为:";
InOrderTraverse(NewT);
cout<<endl;
cout<<"\n原二叉树后序遍历序列为:";
PostOrderTraverse(T);
cout<<endl;
cout<<"新二叉树后序遍历序列为:";
PostOrderTraverse(NewT);
cout<<endl;
cout<<"\n复制二叉树成功!\n"<<endl;
break;
case 4:
cout<<"\n二叉树的深度为:"<<Depth(T)<<"\n"<<endl;
break;
case 5:
cout<<"\n二叉树的结点数为:"<<NodeCount(T)<<"\n"<<endl;
break;
case 6:
cout<<"\n二叉树的叶子结点数为:"<<LeafCount(T)<<"\n"<<endl;
break;
case 0:
exit(0);
}
system("pause");
}
return 0;
}
五、哈夫曼树及其编码
- 这一章的另一个重点算法是关于哈夫曼树的生成与编码。哈夫曼树又称最优树,是一类带权路径长度最短的树,在实际应用中有广泛的用途。这里用一个实际例子来描述哈夫曼树的基本操作。
#include <bits/stdc++.h>
using namespace std;
typedef char **HuffmanCode;
typedef struct
{
int weight;
char c;
int parent,lchild,rchild;
}HTNode,*HuffmanTree;
bool cmp(char a,char b)
{
return a<b;
}
void Select(HuffmanTree &HT,int x,int &s1,int &s2)
{
int i,min1=0x3f3f,min2=0x3f3f;
for (i=1;i<=x;i++)
{
if (HT[i].weight<min1&&HT[i].parent==0)
{
min1=HT[i].weight;
s1=i;
}
}
for (i=1;i<=x;i++)
{
if (HT[i].weight<min2&&i!=s1&&HT[i].parent==0)
{
min2=HT[i].weight;
s2=i;
}
}
}
void CreateHuffmanTree(HuffmanTree &HT,string s,int n)
{
if (n<=1)
return;
int m=2*n-1;
HT=new HTNode[m+1];
for (int i=1;i<=m;i++)
{
HT[i].parent=0;
HT[i].lchild=0;
HT[i].rchild=0;
}
sort(s.begin(),s.end(),cmp);
map<char,int> mp;
for (int i=0;i<s.length();i++)
mp[s[i]]++;
map<char,int>::iterator it;
int j=1;
int a[30];
memset(a,0,sizeof(a));
for (it=mp.begin();it!=mp.end();it++)
{
cout<<it->first<<":"<<it->second<<endl;
a[j++]=it->second;
}
for (int i=1;i<=n;i++)
HT[i].weight=a[i];
for (int i=n+1;i<=m;i++)
{
int s1,s2;
Select(HT,i-1,s1,s2);
HT[s1].parent=i;
HT[s2].parent=i;
HT[i].lchild=s1;
HT[i].rchild=s2;
HT[i].weight=HT[s1].weight+HT[s2].weight;
}
cout<<"哈夫曼树终态:"<<endl;
cout<<"i\tw\tp\tl\tr"<<endl;
for (int i=1;i<=m;i++)
cout<<i<<"\t"<<HT[i].weight<<"\t"<<HT[i].parent<<"\t"<<HT[i].lchild<<"\t"<<HT[i].rchild<<endl;
}
void CreateHuffmanCode(HuffmanTree HT,HuffmanCode &HC,string s,int n)
{
HC=new char*[n+1];
char *cd=new char[n+1];
cd[n-1]='\0';
for (int i=1;i<=n;i++)
{
int start=n-1;
int c=i;
int f=HT[i].parent;
while (f!=0)
{
--start;
if (HT[f].lchild==c)
cd[start]='0';
else
cd[start]='1';
c=f;
f=HT[f].parent;
}
HC[i]=new char[n-start];
strcpy(HC[i],&cd[start]);
}
delete cd;
}
void Decoding(HuffmanTree &HT,HuffmanCode &HC,string t,int n)
{
int m=2*n-1;
int i=0;
while (t[i]!='\0')
{
if (t[i]=='0')
m=HT[m].lchild;
else if (t[i]=='1')
m=HT[m].rchild;
if (HT[m].lchild==0)
{
cout<<HT[m].c;
m=2*n-1;
}
i++;
}
cout<<endl;
}
int main()
{
HuffmanTree HT;
HuffmanCode HC;
string s;
cout<<"请输入电文:";
cin>>s;
set<char> S;
for (int i=0;i<s.length();i++)
S.insert(s[i]);
int n=S.size();
CreateHuffmanTree(HT,s,n);
CreateHuffmanCode(HT,HC,s,n);
cout<<endl;
set<char>::iterator it;
int i=1;
for (it=S.begin();it!=S.end();it++)
HT[i++].c=*it;
for (int i=1;i<=n;i++)
cout<<HT[i].c<<"的哈夫曼编码为:"<<HC[i]<<endl;
string t="";
cout<<"电文编码为:";
for (int i=0;i<s.length();i++)
{
for (int j=1;j<=n;j++)
{
if (s[i]==HT[j].c)
{
cout<<HC[j];
t+=HC[j];
}
}
}
cout<<endl;
cout<<"电文解码为:";
Decoding(HT,HC,t,n);
return 0;
}
哈夫曼树的操作如上,在这个算法中,看似很容易,实际上是比较的复杂,它涉及到了很多的知识,在读这一部分的代码时,大家只需要掌握哈夫曼树的构造和哈夫曼编码的构造过程即可,解码部分作为了解即可。