内容:编写一个程序实现二叉树的基本运算,完成如下功能:
(1)由如图7.1所示的二叉树创建对应的二叉链存储结构b,该二叉树的括号表示串为“A(B(D,E(H(J,K(L,M(,N))))),C(F,G(,I)))”。
(2)输出二叉树b。
(3)输出‘H'结点的左、右孩子结点值。
(4)输出二叉树6的高度。
(5)释放二叉树b。
值得注意的是二叉树的创建相比之前难度要大,这里给出详细解释:
1. BTnode* st[max] 定义了一个指针数组st,用于存储二叉树节点的指针。
2. *p 定义了一个指向BTnode类型的指针p,用于指向当前处理的节点。
3. int top = -1 定义了一个整型变量top,用于表示栈顶位置,初始值为-1。
4. int k 定义了一个整型变量k,用于表示当前处理的节点是左子树还是右子树。
5. char ch 定义了一个字符变量ch,用于存储当前处理的字符。
这些变量和数组在createBTree函数中用于创建二叉树,通过栈的方式实现了二叉树的创建过程。
代码如下:
#include<stdio.h>
#include<stdlib.h>
#define max 100
typedef char elemtype;
typedef struct node
{
elemtype data;
struct node* lchild;
struct node* rchild;
}BTnode; //声明二叉链节点类型
void createBTree(BTnode*& b, const char* str) //创建二叉树
{
BTnode* st[max], * p;
p = NULL;
int top = -1, k;
char ch;
b = NULL; //建立二叉树初始时为空
ch = str[0];
int j = 0;
while (ch != '\0') //\0为转义字符,意思是空字符,(ch != '\0') 即为ch不为空的意思
{
switch (ch)
{
case'(': top++; st[top] = p; k = 1; break; //开始处理左子树
case')': top--; break; //子树处理完毕
case',': k = 2; break; //开始处理右子树
default: p = (BTnode*)malloc(sizeof(BTnode));
p->data = ch, p->lchild = p->rchild = NULL;
if (b == NULL)
{
b = p;
}
else {
switch (k)
{
case '1': st[top]->lchild = p; break;
case '2': st[top]->rchild = p; break;
}
}
}
j++; ch = str[j];
}
}
void destroyBTree(BTnode*& b) //销毁二叉树(递归)
{
if (b != NULL)
{
destroyBTree(b->lchild);
destroyBTree(b->rchild);
free(b);
}
}
BTnode* findnode(BTnode* b, elemtype x) //查找值为x的结点(递归)
{
BTnode* p;
if (b == NULL)
{
return NULL;
}
else if (b->data == x)
{
return b;
}
else
{
p = findnode(b->lchild, x);
if (p != NULL)
{
return p;
}
else
return findnode(b->rchild, x);
}
}
BTnode* lchildnode(BTnode* p) //返回p结点的左孩子结点指针
{
return p->lchild;
}
BTnode* rchildnode(BTnode* p)
{
return p->rchild;
}
int BTheight(BTnode * b) //求二叉树b的高度
{
int lchildh, rchildh;
if (b == NULL)
{
return 0;
}
else
{
lchildh = BTheight(b->lchild);
rchildh = BTheight(b->rchild);
return (lchildh > rchildh) ? (lchildh + 1) : (rchildh + 1);
}
}
void dispBTnode(BTnode * p) //输出p结点的值
{
if (p != NULL)
{
printf("%c", p->data);
}
}
void dispBTtree(BTnode * b) //输出二叉树b
{
if (b != NULL)
{
printf("%c", b->data);
if (b->lchild != NULL || b->rchild != NULL)
{
printf("(");
dispBTtree(b->lchild);
if (b->rchild != NULL)
{
printf(",");
}
dispBTtree(b->rchild);
printf(")");
}
}
}
int main()
{
BTnode* b;
createBTree(b, "A(B(D,E(H(J,K(L,M(,N)))),C(F,G(,I)))");
printf("二叉树的基本运算如下\n");
printf("1.创建二叉树\n");
printf("2.输出二叉树:");
dispBTtree(b);
printf("\n");
BTnode* h = findnode(b, 'H');
printf("3.H结点:");
if (h != NULL)
{
BTnode* lp = lchildnode(h);
if (lp != NULL)
{
printf("左孩子为%c", lp->data);
}
else
{
printf("无左孩子");
}
BTnode* rp = rchildnode(h);
if (rp != NULL)
{
printf("右孩子为%c", rp->data);
}
else
{
printf("无右孩子");
}
}
printf("\n");
printf("4.二叉树b的高度:%d\n", BTheight(b));
printf("5.释放二叉树b\n");
destroyBTree(b);
return 0;
}
运行结果如下: