中序遍历二叉树算法思想:①沿着根的左孩子依次入栈,直到左孩子为空。②栈顶元素出栈并访问,若其右孩子为空,则继续执行②;③若右孩子不为空,则执行步骤①。
先序遍历二叉树算法思想:①沿着根的左孩子依次访问并入栈,直到左孩子为空。②栈顶元素出栈,若其右孩子为空,则继续执行②;③若右孩子不为空,则执行步骤①。
后序遍历二叉树算法思想:①沿着根节点的左孩子,依次入栈,直到左孩子为空;②读取栈顶元素:若其右孩子不为空,且未被访问过,则右子树转向执行①;否则,栈顶元素出栈并访问。
(1)func.h
#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
//二叉树相关数据结构
typedef char ElemType;
typedef struct BiTNode {
ElemType data;
struct BiTNode* lchild, * rchild;
}BiTNode,*BiTree;
//先序建二叉树
void creatBiTree(BiTree &T);
//访问当前结点
void visit(BiTNode* pnode);
//中序遍历二叉树
/*算法思想:①沿着根的左孩子依次入栈,直到左孩子为空。②栈顶元素出栈并访问,若其右孩子为空,则继续执行②;③若右孩子不为空,则执行步骤①*/
void inOrder(BiTree T);
//先序遍历二叉树
/*算法思想:①沿着根的左孩子依次访问并入栈,直到左孩子为空。②栈顶元素出栈,若其右孩子为空,则继续执行②;③若右孩子不为空,则执行步骤①*/
void preOrder(BiTree T);
//后序遍历二叉树
/*算法思想:①沿着根节点的左孩子,依次入栈,直到左孩子为空;②读取栈顶元素:若其右孩子不为空,且未被访问过,则右子树转向执行①;
否则,栈顶元素出栈并访问*/
void postOrder(BiTree T);
//链栈相关数据结构
typedef BiTree SElemType;
typedef struct LinkNode {
SElemType data;
struct LinkNode* next;
}LinkNode,*LinkStack;
//初始化栈
void initial(LinkStack& S);
//判栈空
int isEmpty(LinkStack S);
//插入链头
void Push(LinkStack& S, SElemType x);
//重链头出栈
void Pop(LinkStack& S, SElemType& x);
//读取栈顶元素
void GetTop(LinkStack S, SElemType& x);
(2)Stack.cpp
#include"func.h"
//初始化栈
void initial(LinkStack& S) {
S = (LinkNode*)malloc(sizeof(LinkNode));
S->next = NULL;
}
//判栈空
int isEmpty(LinkStack S) {
if (S->next == NULL) {
return 1;
}
else {
return 0;
}
}
//插入链头
void Push(LinkStack& S, SElemType x) {
LinkNode* pnode = (LinkNode*)malloc(sizeof(LinkNode));
pnode->data = x;
pnode->next = S->next;
S->next = pnode;
}
//重链头出栈
void Pop(LinkStack& S, SElemType &x) {
LinkNode* pnode = S->next;
x = pnode->data;
S->next = pnode->next;
free(pnode);
pnode = NULL;
}
//读取栈顶元素
void GetTop(LinkStack S, SElemType& x) {
x = S->next->data;
}
(3)BiTree.cpp
#include"func.h"
//先序建二叉树
void creatBiTree(BiTree &T) {
ElemType ch;
scanf("%c", &ch);
if (ch == '#') {
T = NULL;
}
else {
T = (BiTNode*)malloc(sizeof(LinkNode));
T->data = ch;
creatBiTree(T->lchild);
creatBiTree(T->rchild);
}
}
//访问当前结点
void visit(BiTNode* pnode) {
printf("%c", pnode->data);
}
//中序遍历二叉树
/*算法思想:①沿着根的左孩子依次入栈,直到左孩子为空。②栈顶元素出栈并访问,若其右孩子为空,则继续执行②;③若右孩子不为空,则执行步骤①*/
void inOrder(BiTree T) {
LinkStack S = NULL;
initial(S);
BiTNode* pnode = T;
while (pnode != NULL || !isEmpty(S)) {//栈不为空或者当前结点不为空时循环
if (pnode) {//沿着根节点的左孩子入栈,直到左孩子为空
Push(S, pnode);
pnode = pnode->lchild;
}
else {//栈顶元素出栈并访问,然后转向他的右子树
Pop(S, pnode);
visit(pnode);
pnode = pnode->rchild;
}
}
}
//先序遍历二叉树
/*算法思想:①沿着根的左孩子依次访问并入栈,直到左孩子为空。②栈顶元素出栈,若其右孩子为空,则继续执行②;③若右孩子不为空,则执行步骤①*/
void preOrder(BiTree T) {
LinkStack S;
initial(S);
BiTNode* pnode = T;
while (pnode || !isEmpty(S)) {
if (pnode) {
visit(pnode);
Push(S, pnode);
pnode = pnode->lchild;
}
else {
Pop(S, pnode);
pnode = pnode->rchild;
}
}
}
//后序遍历二叉树
/*算法思想:①沿着根节点的左孩子,依次入栈,直到左孩子为空;②读取栈顶元素:若其右孩子不为空,且未被访问过,则右子树转向执行①;
否则,栈顶元素出栈并访问*/
void postOrder(BiTree T) {
LinkStack S = NULL;
initial(S);
BiTNode* pnode= T;
BiTNode* rnode = NULL;
while (pnode || !isEmpty(S)) {
if (pnode) {//沿着根节点的左孩子,依次入栈,直到左孩子为空
Push(S, pnode);
pnode = pnode->lchild;
}
else {
GetTop(S, pnode);//读取栈顶元素
if (pnode->rchild && pnode->rchild != rnode) {//若其右孩子不为空,且未被访问过,则右子树转向执行①
pnode = pnode->rchild;
}
else {//栈顶元素出栈并访问
Pop(S, pnode);
visit(pnode);
rnode = pnode;//记录最近访问过的结点
pnode = NULL;//结点访问完后,重置p指针为空(注意:每次出栈访问完一个结点就相当于遍历完以该节点为根的子树,需要将p置NULL)
}
}
}
}
(4)main.cpp
#include"func.h"
//AB#D##CE###
//BDAEC(中序)
//ABDCE(先序)
//DBECA(后序)
int main() {
BiTree T = NULL;
creatBiTree(T);
inOrder(T);
printf("\n");
preOrder(T);
printf("\n");
postOrder(T);
printf("\n");
return 0;
}