2024王道408数据结构 P144 T9
方法一:非递归
思考过程
- 这题跟上一题一样,只需要入队的时候先判断有无右子树,有的话入队,再判断左子树,有的话入队就行了,如果没有右子树就判断有无左子树,有左子树就入队。
- 首先我们还是需要一个辅助队来帮我们遍历二叉树,还需要一个指针p。
举个例子
- 假设这个二叉树长这样
首先我们把队列给初始化,在循环开始之前先把根结点也就是A结点入队,入队之后开始判断是要队列不满就进入循环while (!QueueEmpty(q))
- 进入循环第一步就是先把队首结点给出队,出队之后此时指针p是指向结点A的,先判断A是否有右子树,有的话就入队,再去判断是否有左子树,有的话入队,在上面的例子里就是先入队C,在入队B
此时指针还是指向A的,到这里第一次循环就完成了。 - 第二次循环开始时第一步先把队首结点给出队,也就是现在的C,此时指针p是指向C的,再去判断C有无右子树和左子树,有的话一起入队,注意一定要是先判断右子树再判断左子树,到这里我们就把G和F都入队了
- 接下来再出队结点B,就这样循环往复,写到这里这题就做完了。
这种方法就是把先序遍历换一下,先遍历根结点,再遍历右子树,最后是左子树。
方法二:递归
- 递归就很好写,首先例子还是上面方法一那个例子,先用一个指针p把左子树给存起来,再把右子树的值赋给左子树,最后再把p的值赋给右子树。
- 举个例子先根结点A传进去,递归再传入A的左子树B,再传入B的左子树D,此时D没有左右子树了,也就不执行左子树的递归和右子树的递归了。假装D有左右子树,那么这次递归就是把D的左右子树交换了(但是实际上D并没有左右子树),再往上它的双亲到B,B再去递归他的右子树E,把E的左右子树也交换一遍。再轮到结点B,把B的左右子树交换。
- C也是同理,递归完C之后再轮到A,把A的左右子树交换。
完整代码
#include "iostream"
#define MAX 15
struct TreeNode
{
char data;
struct TreeNode *lchild, *rchild;
};
struct Queue
{
struct TreeNode *data[MAX];
int f, r, tag;
};
bool QueueEmpty(Queue q)
{
if (q.r == q.f && q.tag == 0)
return true;
return false;
}
bool QueueOverflow(Queue q)
{
if (q.r == q.f && q.tag == 1)
return true;
return false;
}
bool EnQueue(Queue &q, struct TreeNode *&p)
{
if (QueueOverflow(q))
{
printf("队满,入队失败\n");
return false;
}
else
{
q.data[q.r] = p;
q.r = (q.r + 1) % MAX;
q.tag = 1;
return true;
}
}
bool DeQueue(Queue &q, struct TreeNode *&p)
{
if (QueueEmpty(q))
{
printf("队空,入队失败\n");
return false;
}
else
{
p = q.data[q.f];
q.f = (q.f + 1) % MAX;
q.tag = 0;
return true;
}
}
void display(struct TreeNode *t)
{
if (t)
{
printf("%c ", t->data);
display(t->lchild);
display(t->rchild);
}
}
void CreateTree(struct TreeNode *&t)
{
char ch = getchar();
if (ch == '#')
t = NULL;
else
{
t = (struct TreeNode*)malloc(sizeof(struct TreeNode));
t->data = ch;
t->lchild = NULL;
t->rchild = NULL;
CreateTree(t->lchild);
CreateTree(t->rchild);
}
}
void reverse1(struct TreeNode *&t)
{
Queue q;
q.r = q.f = q.tag = 0;
struct TreeNode *p = t;
EnQueue(q, p);
printf("%c ",p->data);
while (!QueueEmpty(q))
{
DeQueue(q, p);
if (p->rchild)
{
EnQueue(q, p->rchild);
if (p->lchild)
EnQueue(q, p->lchild);
}
else
{
if (p->lchild)
EnQueue(q, p->lchild);
}
}
}
void reverse2(struct TreeNode *&t)
{
struct TreeNode *p;
if (t)
{
reverse2(t->lchild);
reverse2(t->rchild);
p = t->lchild;
t->lchild = t->rchild;
t->rchild = p;
}
}
int main()
{
struct TreeNode *t;
CreateTree(t);
display(t);
printf("\n");
printf("方法二交换后:");
reverse2(t);
display(t);
return 0;
}