C++实现先序创建二叉树,并用递归和非递归两种方式二叉树中序遍历
这是数据结构课堂留下的一个作业:
题目是:按先序次序输入二叉树中的结点的值(data为字符类型),以’#'表示空字符,创建二叉链表示二叉树。函数原型如下:
void CreateBiTree(BiTree *T);//创建二叉树函数
void InOrderTraverse(BiTree T); //利用栈以非递归的方式实现中序遍历上述二叉树
二叉链表的定义如下:
typedef struct BiTNode{
char data;
struct BiNode *lchild,*rchild;
}BiTNode,*BiTree; //这里将BiTNode和BiTree声明为struct BiTNode和struct BiTNode*的别名
代码运行实例:
于是按照题目的要求写了第一个版本的代码(问题主要出现在二叉树的创建上,这里先不贴中序遍历的代码):
#include<iostream>
#include<malloc.h>
using namespace std;
//定义二叉链表的结构体
typedef struct BiTNode {
char data; //用来存储数据
struct BiTNode* lchild, * rchild; //左右子树
}BiTNode, * BiTree;
//函数声明
void CreateBiTree(BiTree *T);
int main() {
cout << "请输入建立二叉链表的序列:" << endl;
BiTree T = 0; //初始化为空指针
CreateBiTree(&T);
//cout << T->data;
return 0;
}
void CreateBiTree(BiTree *T) {
char ch;
cin >> ch;
if (ch == '#') {
*T = NULL;
}
else {
*T = (BiTNode*)malloc(sizeof(BiTNode));
(*T)->data = ch;
CreateBiTree(&(*T)->lchild);
CreateBiTree(&(*T)->rchild);
}
}
在上述的代码中,由于题目中传给 CreateBiTree(BiTree *T) 函数的参数是指向 BiTree 的指针,而根据 Typedef 的定义 BiTree 本身就是指向 struct BiTNode 的指针类型,于是 BiTree *T 就成为了指向 struct BiTNode 的指针的指针,所以创建二叉树的代码就变成了 (*T)->data = ch 和 CreateBiTree(&(*T)->lchild) 风格,虽然能够运行,但是有种“多此一举”的感觉。(这是因为C语言中没有引用,而创建二叉树需要修改参数BiTree T本身,所以就需要使用BiTree *T作为参数)。
于是出现了第二个版本:
CreateBiTree(BiTree T);//调用函数
void CreateBiTree(BiTree T) {
char ch;
cin >> ch;
if (ch == '#') {
T = NULL;
}
else {
T = (BiTNode*)malloc(sizeof(BiTNode));
T->data = ch;
CreateBiTree(T->lchild);
CreateBiTree(T->rchild);
}
}
直接将 BiTree 作为函数的参数,虽然这样做程序能够运行,但是 BiTree T 作为实参调用 CreateBiTree() 函数后,其仍然为空指针,也就是说二叉树没有建立成功,原因是此时的形参是 BiTree 类型的变量,实参也是 BiTree 类型的变量,而用指针作为函数参数时能够改变的是指针所指之物,而无法改变指针本身,因此二叉树没有建立成功。
最后的版本是将 BiTree 的引用作为函数的形参,而引用作为形参时,其值的改变可以引起实参值的改变。此时二叉树成功的建立了。
CreateBiTree(T); //调用函数
void CreateBiTree(BiTree &T) {
char ch;
cin >> ch;
if (ch == '#') {
T = NULL;
}
else {
T = (BiTNode*)malloc(sizeof(BiTNode));
T->data = ch;
CreateBiTree(T->lchild);
CreateBiTree(T->rchild);
}
}
最后附上完整的代码:
#include<iostream>
#include<malloc.h>
#include<stack>
using namespace std;
//定义二叉链表的结构体
typedef struct BiTNode {
char data; //用来存储数据
struct BiTNode* lchild, * rchild; //左右子树
}BiTNode, * BiTree;
//函数声明
void CreateBiTree(BiTree &T);
void InOrderTraverse(BiTree T);
int main() {
cout << "请输入建立二叉链表的序列:" << endl;
BiTree T = 0; //初始化为空指针
CreateBiTree(T);
cout << "中序遍历的结果为:" << endl;
InOrderTraverse(T);
return 0;
}
//递归方式实现先序二叉树的创建
void CreateBiTree(BiTree &T) {
char ch;
cin >> ch;
if (ch == '#') {
T = NULL;
}
else {
T = (BiTNode*)malloc(sizeof(BiTNode));
T->data = ch;
CreateBiTree(T->lchild);
CreateBiTree(T->rchild);
}
}
//利用递归的方式二叉树的中序遍历
void InOrderTraverse(BiTree T){
if(T==NULL)
return OK;
else{
InOrderTraverse(T->lchild);
cout << T->data;
InOrderTraverse(T->rchild);
}
}
//利用栈实现非递归二叉树中序遍历
void InOrderTraverse(BiTree T) {
stack<BiTree> S; //初始化栈
BiTree p = T;
while (p || !S.empty()) {
//根指针进栈,遍历左子树
if (p) {
S.push(p);
p = p->lchild;
}
//根指针退栈,访问根结点,遍历右子树
else {
p = S.top();
S.pop();
cout << p->data;
p = p->rchild;
}
}
}