【天梯赛】L2-011 玩转二叉树 (25分)

不念不忘少年蓝@ 2022-12-20 14:58 175阅读 0赞

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

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

**模板题吧，考察点还是两个：根据前序和中序遍历还原二叉树；层序遍历输出。**

## 题目说左右节点相交换，我的第一思路是按照题目说的去依次翻转，然后将得到的二叉树在进行层序遍历输出。但很快我感觉，只要将左右节点的入队顺序反过来，出队时就是翻转的。嗯，果然如此 ##

#include <iostream>
    #include <queue>
    using namespace std;
    int pre[32], in[32];
    typedef struct bitnode{
        
    	int data;
    	struct bitnode *lchild, *rchild;
    }BiTNode, *BiTree;
    
    BiTree buildTree(int root, int start, int end){
        
    	if(start > end) return NULL;
    	BiTree p = (BiTNode *)malloc(sizeof(BiTNode));
    	p->data = pre[root];
    	int i, cnt = 0;
    	for(i = start; i < end; i++){
        
    		cnt++;
    		if(in[i] == pre[root])
    			break;
    	}
    	p->lchild = buildTree(root+1, start, i-1);
    	p->rchild = buildTree(root+cnt, i+1, end);
    	return p;
    }
    
    queue<BiTree> q;
    void Levelorder(BiTree bt){
        
    	q.push(bt);
    	int num = 0;
    	while(!q.empty()){
        
    		BiTree tmp = q.front();
    		if(num > 0) cout << " ";
    		cout << tmp->data;
    		if(tmp->rchild) q.push(tmp->rchild);
    		if(tmp->lchild) q.push(tmp->lchild);
    		q.pop();
    		num++;
    	}
    }
    int main(){
        
    	int n;
    	cin >> n;
    	for(int i = 0; i < n; i++){
        
    		cin >> in[i];
    	}
    	for(int i = 0; i < n; i++){
        
    		cin >> pre[i];
    	}
    	BiTree p = buildTree(0, 0, n-1);
    	Levelorder(p);
    }