数据结构——树和二叉树的操作实现

数据结构——树和二叉树的操作实现

  • 树和二叉树这一章的知识点相较于前几章而言有一点难度,这一章的算法都是以递归的形式呈现的,对算法的理解要求较高。这一章的操作实现以二叉树的操作为主。

一、二叉树的结构定义

  树的存储结构可以是顺序存储,也可以链式存储。这一章的算法以链式的为主,所以在这里给出二叉树的链式存储结构定义。这里的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;
}

  哈夫曼树的操作如上,在这个算法中,看似很容易,实际上是比较的复杂,它涉及到了很多的知识,在读这一部分的代码时,大家只需要掌握哈夫曼树的构造和哈夫曼编码的构造过程即可,解码部分作为了解即可。

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值