void LDRTree(CBTType *treeNode)
{
if(treeNode)
{
LDRTree(treeNode->left); //顯示左子樹內容
ShowNodeData(treeNode); //顯示結點內容
DLRTree(treeNode->right); //顯示右子樹內容
}
}
#include<iostream>
using namespace std;
#define MAXLEN 20 //最大長度
typedef char DATA; //定義元素類型
struct CBTType /定義二叉樹結點類型
{
DATA data; //元素數據
CBTType * left; //左子樹結點指針
CBTType * right; //右子樹結點指針
};
/*********************初始化二叉樹***********************/
CBTType * InitTree()
{
CBTType * node;
if(node = new CBTType) //申請內存
{
cout<<"請先輸入一個根節點數據："<<endl;
cin>>node->data;
node->left=NULL;
node->right=NULL;
if(node!=NULL) //如果二叉樹結點不為空
{
return node;
} else
{
return NULL;
}
}
return NULL;
}
/***********************查找結點*************************/
CBTType *TreeFindNode(CBTType *treeNode,DATA data)
{
CBTType *ptr;
if(treeNode==NULL)
{
return NULL;
}else
{
if(treeNode->data==data)
{
return treeNode;
}
else //分別向左右子樹查找
{
if(ptr=TreeFindNode(treeNode->left,data)) //左子樹遞歸查找
{
return ptr;
}
else if(ptr=TreeFindNode(treeNode->right,data)) //右子樹遞歸查找
{
return ptr;
}
else
{
return NULL;
}
}
}
}
/**********************添加結點*************************/
void AddTreeNode(CBTType *treeNode)
{
CBTType *pnode,*parent;
DATA data;
char menusel;
if(pnode=new CBTType) //分配內存
{
cout<<"輸入二叉樹結點數據："<<endl;
cin>>pnode->data;
pnode->left=NULL; //設置左子樹為空
pnode->right=NULL; //設置左子樹為空
cout<<"輸入該結點的父結點數據"<<endl;
cin>>data;
parent=TreeFindNode(treeNode,data); //查找父結點，獲得結點指針
if(!parent)
{
cout<<"沒有找到父結點"<<endl;
delete pnode;
return ;
}
cout<<"1.添加該結點到左子樹；2.添加該結點到右子樹。請輸入操作對應的數字。"<<endl;
do
{
cin>>menusel;
if(menusel=='1'||menusel=='2')
{
switch(menusel)
{
case '1': //添加結點到左子樹
if(parent->left) //左子樹不為空
{
cout<<"左子樹結點不為空"<<endl;
}
else
{
parent->left=pnode;
cout<<"數據添加成功！"<<endl;
}
break;
case '2': //添加結點到右子樹
if(parent->right) //右子樹不為空
{
cout<<"右子樹結點不為空"<<endl;
}
else
{
parent->right=pnode;
cout<<"數據添加成功！"<<endl;
}
break;
default:
cout<<"子節點選擇error!"<<endl;
break;
}
}
}while(menusel!='1'&&menusel!='2');
}
}
/***********************計算二叉樹的深度********************************/
int TreeDepth(CBTType *treeNode)
{
int depleft,depright;
if(treeNode==NULL)
{
return 0; //結點為空的時候，深度為0
}
else
{
depleft=TreeDepth(treeNode->left); //左子樹深度（遞歸調用）
depright=TreeDepth(treeNode->right); //右子樹深度（遞歸調用）
if(depleft)
{
return ++depleft;
}
else
{
return ++depright;
}
}
}
/*************************顯示結點數據*********************************/
void ShowNodeData(CBTType *treeNode)
{
cout<<treeNode->data<<"/t"; //直接輸出結點數據
}
/***********************清空二叉樹************************************/
void ClearTree(CBTType *treeNode)
{
if(treeNode) //判斷當前樹不為空
{
ClearTree(treeNode->left); //清空左子樹
ClearTree(treeNode->right); //清空右子樹
delete treeNode; //釋放當前結點所占用的內存
}
}
/**************************按層遍歷算法*********************************/
void LevelTree(CBTType *treeNode)
{
CBTType *p;
CBTType *q[MAXLEN]; //定義一個隊列
int head=0,tail=0;
if(treeNode) //如果隊首指針不為空
{
tail=(tail+1)%MAXLEN; //計算循環隊列隊尾序號
q[tail]=treeNode; //二叉樹根指針進入隊列
while(head!=tail)
{
head=(head+1)%MAXLEN; //計算循環隊列的隊首序號
p=q[head]; //獲取隊首元素
ShowNodeData(p); //輸出隊首元素
if(p->left!=NULL) //如果存在左子樹
{
tail=(tail+1)%MAXLEN; //計算隊列的隊尾序號
q[tail]=p->left; //左子樹入隊
}
if(p->right!=NULL) //如果存在右子樹
{
tail=(tail+1)%MAXLEN; //計算隊列的隊尾序號
q[tail]=p->right; //右子樹入隊
}
}
}
}
/*************************先序遍歷算法**********************************/
void DLRTree(CBTType *treeNode)
{
if(treeNode)
{
ShowNodeData(treeNode); //顯示結點內容
DLRTree(treeNode->left); //顯示左子樹內容
DLRTree(treeNode->right); //顯示右子樹內容
}
}
/***********************中序遍歷算法************************************/
void LDRTree(CBTType *treeNode)
{
if(treeNode)
{
LDRTree(treeNode->left); //顯示左子樹內容
ShowNodeData(treeNode); //顯示結點內容
DLRTree(treeNode->right); //顯示右子樹內容
}
}
/***********************后序遍歷算法************************************/
void LRDTree(CBTType *treeNode)
{
if(treeNode)
{
LRDTree(treeNode->left); //顯示左子樹內容
DLRTree(treeNode->right); //顯示右子樹內容
ShowNodeData(treeNode); //顯示結點內容
}
}
/*************************主函數部分************************************/
int main()
{
CBTType *root=NULL; //root為指向二叉樹根結點的指針
char menusel;
//設置根結點
root=InitTree();
//添加結點
do
{
cout<<"請選擇菜單添加二叉樹的結點："<<endl;
cout<<"0.退出;1.添加二叉樹的結點。"<<endl;
cin>>menusel;
switch(menusel)
{
case '1':
AddTreeNode(root);
break;
case '0':
break;
default:
cout<<"添加結點error"<<endl;
break;
}
}while(menusel!='0');
//輸出樹的深度
cout<<"depth:"<<TreeDepth(root)<<endl;
//輸出結點內容
do
{
cout<<"請選擇菜單遍歷二叉樹，輸入0表示退出："<<endl;
cout<<"1.按層遍歷"<<endl;
cout<<"2.先序遍歷DLR"<<endl;
cout<<"3.中序遍歷LDR"<<endl;
cout<<"4.后序遍歷LRD"<<endl;
cin>>menusel;
switch(menusel)
{
case '0':break;
case '1':
cout<<"按層遍歷的結果："<<endl;
LevelTree(root);
cout<<endl;
break;
case '2':
cout<<"先序遍歷的結果："<<endl;
DLRTree(root);
cout<<endl;
break;
case '3':
cout<<"中序遍歷的結果："<<endl;
LDRTree(root);
cout<<endl;
break;
case '4':
cout<<"后序遍歷的結果："<<endl;
LRDTree(root);
cout<<endl;
break;
default:
cout<<"遍歷方式選擇出錯！"<<endl;
break;
}
}while(menusel!='0');
//清空二叉樹
ClearTree(root);
return 0;
}
