L2-011. 玩转二叉树

PATL2-011. 玩转二叉树

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

---

题解：
根据中序遍历（左中右），前序遍历（中左右）；
由每次前序遍历区间的第一一个节点为跟节点，在中序遍历中找到这个节点即可将树的左子树和右子树区分，通过递归不断还原整棵树。

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <queue>
using namespace std ;
struct node
{
    int l , r;
}tree[50];
int pre[50] , in[50]  , ans[50] , n;
int build(int prel , int prer , int inl , int inr)
{
    if(inl > inr) return 0;
    int root = pre[prel] , p1 , p2 ;
    p1 = inl ;
    while(in[p1] != root) p1 ++;
    p2 = p1 - inl;
    tree[root].l = build(prel+1,prel+p2,inl , p1 - 1);
    tree[root].r = build(prel+p2+1,prer,p1+1 , inr);
    return root;
}
void BFS()
{
    queue<int> q ;
    int tt = 0 ;
    q.push(pre[0]);
    int temp ;
    while(!q.empty())
    {
        temp = q.front();
        q.pop();
        if(temp == 0) break ;
        ans[tt++] = temp;
        if(tree[temp].r!= 0)
        {
            q.push(tree[temp].r);
            //ans[tt++] = tree[temp].r ;
        }
        if(tree[temp].l != 0)
        {
            q.push(tree[temp].l);
            //ans[tt++] = tree[temp].l ;
        }
    }
    for(int i = 0 ; i <tt ; i ++)
    {
        if(i == tt-1)
            printf("%d\n" , ans[i]);
        else
            printf("%d " , ans[i]) ;
    }
}
int main()
{
    cin >> n ;
    for(int i = 0 ; i < n ; i ++)
        scanf("%d" , &in[i]) ;
    for(int i = 0 ; i < n ; i++)
        scanf("%d" , &pre[i]) ;
    build(0 , n-1 , 0 , n-1);
    BFS();
    return 0;
}