L2-011. 玩转二叉树-PAT团体程序设计天梯赛GPLT

L2-011. 玩转二叉树-PAT团体程序设计天梯赛GPLT

L2-011. 玩转二叉树
给定一棵二叉树的中序遍历和前序遍历，请你先将树做个镜面反转，再输出反转后的层序遍历的序列。所谓镜面反转，是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数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

分析：与前序中序转换为后序的代码相仿（无须构造二叉树再进行广度优先搜索~~），只不过加一个变量index，表示当前的根结点在二叉树中所对应的下标（从0开始），所以进行一次输出先序的递归的时候，就可以把根结点下标所对应的值存储在level数组中（一开始把level都置为-1表示此处没有结点），镜面反转只需改变index的值，左孩子为2 * index + 2， 右孩子为2 * index + 1，这样在递归完成后level数组中非-1的数就是按照下标排列的层序遍历的顺序~

#include <cstdio>#include <vector>using namespace std;vector<int> in, pre, level(100000, -1);void levelorder(int root, int start, int end, int index){    if(start > end) return ;    int i = start;    while(i < end && in[i] != pre[root]) i++;    level[index] = pre[root];    levelorder(root + 1, start, i - 1, 2 * index + 2);    levelorder(root + 1 + i - start, i + 1, end, 2 * index + 1);}int main(){    int n, cnt = 0;    scanf("%d", &n);    in.resize(n);    pre.resize(n);    for(int i = 0; i < n; i++) scanf("%d", &in[i]);    for(int i = 0; i < n; i++) scanf("%d", &pre[i]);    levelorder(0, 0, n-1, 0);    int k = level.size();    for(int i = 0; i < k; i++)    {        if(level[i] != -1 && cnt != n - 1)        {            printf("%d ", level[i]);            cnt++;        }        else if(level[i] != -1)        {            printf("%d", level[i]);            break;        }    }    return 0;}