L2-3 玩转二叉树 (25分)
给定一棵二叉树的中序遍历和前序遍历,请你先将树做个镜面反转,再输出反转后的层序遍历的序列。所谓镜面反转,是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数N
(≤30),是二叉树中结点的个数。第二行给出其中序遍历序列。第三行给出其前序遍历序列。数字间以空格分隔。
输出格式:
在一行中输出该树反转后的层序遍历的序列。数字间以1个空格分隔,行首尾不得有多余空格。
输入样例:
7
1 2 3 4 5 6 7
4 1 3 2 6 5 7
输出样例:
4 6 1 7 5 3 2
注意 关于要引用传值
注意传递树参数的方式是引用传值。记得我上个星期第一次写这个题,写了半天不对,和书上对比了一下,也找不出来。后来debug发现,似乎和我在大一上写链表的bug一样,记得当时实验课也找老师找了好长时间bug。
比如我们ReBt一个左子树的时候
ReBt(a + 1, idx - c + a, c, idx - 1, D->l);
我们传递的是D->l这个参数,如果不用引用传参,那么只是把D->l传递过去,她只是一个地址。而我们可能这样想,我们把D->l 即他的左子树传递过去了,然后让他的这个左子树(看代码
void ReBt(int a, int b, int c, int d, BT D)//如果是 没有引用传值
{
D = (BT)malloc(sizeof(BinTreeNode));
D->data = pre[a];
让他的这个左子树 = 一个新生成的空间。我们心里这样想,感觉就对了。其实不是的。
我们只是 在递归这个函数(就是第二级函数这),新建了一个指针,指针的值是上个结点的左子树,现在让这个指针重新指向了一个新的地址。那么,和我之前D->l有什么关系(doge)
比如现在D->l 指的是NULL。那么在第二级函数这里有个指针D 他的值是NULL,然后指向了一个新的地址。
有三种方式来解决这个问题。
第一个就是 用指针,对,指向指针的指针,让这个指针指向的地址的那个元素(即D->l 这个元素) 指向malloc…。
第二种就是引用传值,他应该就是封装的指针吧。
第三种就是带上返回值,在第一层函数里让D->l=这个新的地址。
代码
#include <iostream>
#include<deque>
using namespace std;
typedef struct BinTreeNode *BT;
struct BinTreeNode
{
int data;
BT l, r;
};
int mid[30], pre[30];
void ReBt(int a, int b, int c, int d, BT &D);
void ExchangeChild(BT bt);
void PrintBy(BT bt)
{
deque<BT> de;
int cnt=0;
if(bt)
de.push_back(bt);
while(!de.empty())
{
BT tmp=de.front();
de.pop_front();
if(cnt++!=0)
printf(" ");
printf("%d",tmp->data);
if(tmp->l)
de.push_back(tmp->l);
if(tmp->r)
de.push_back(tmp->r);
}
}
int main()
{
BT bt;
int N;
cin >> N;
for (int i = 0; i < N; ++i)
cin >> mid[i];
for (int i = 0; i < N; ++i)
cin >> pre[i];
ReBt(0, N - 1, 0, N - 1, bt);
ExchangeChild(bt);
PrintBy(bt);
}
void ReBt(int a, int b, int c, int d, BT &D)
{
D = (BT)malloc(sizeof(BinTreeNode));
D->data = pre[a];
int idx;
for (idx = c; idx <= d; ++idx)
{
if (mid[idx] == pre[a]) //中序的跟结点
break;
}
if (idx != c) //有左子树
ReBt(a + 1, idx - c + a, c, idx - 1, D->l);
else
D->l = NULL;
if (idx != d) //有右子树
ReBt(idx - c + a + 1, b, idx + 1, d, D->r);
else
D->r = NULL;
}
void ExchangeChild(BT bt)
{
if(bt->l&&bt->r)//not leaf
swap(bt->l,bt->r);
if(bt->l)
ExchangeChild(bt->l);
if(bt->r)
ExchangeChild(bt->r);
}