玩转二叉树（c++）

7-11 玩转二叉树

分数 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

#include <bits/stdc++.h>
using namespace std;
#define iloves	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define int long long
const int MAX_N = 31;
/*二叉树的结构体*/
struct TreeNode{
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x): val(x),left(nullptr),right(nullptr){}//这是一个构造函数 ，使得在创建节点时，给各个变量赋初值；
};
int n,cnt,temp;
int preOrder[MAX_N],inOrder[MAX_N],check_inOrder[MAX_N];
//已知前中序遍历二叉树，建树函数
TreeNode* build(TreeNode  *root){
	int t=preOrder[temp++];
	int k;//这个k只能定义在局部
	for(k=1;k<=n;k++)if(inOrder[k]==t)break;
	root=new TreeNode(t);				//这里要重新建一个root
	check_inOrder[k]=1;
	if(k>1&&k<=n&&check_inOrder[k-1]==0){root->left=build(root->left);}
	if(k>0&&k<n&&check_inOrder[k+1]==0){root->right=build(root->right);}
	return root;
}
//题目要求的反转，只需要在遍历过程中，改变层序遍历顺序，及右子树先入队
void levelOrder(TreeNode* root){
	queue<TreeNode*>q;
	q.push(root);
	while(!q.empty()){
		TreeNode* P=q.front();
		q.pop();
		if(P->val!=root->val)cout<<" "<<P->val;
		else cout<<P->val;
		if(P->right!=nullptr)q.push(P->right);
		if(P->left!=nullptr)q.push(P->left);
		
	}
	cout<<endl;
}
void slove() {
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>cnt;
		inOrder[i]=cnt;
	}
	for(int i=1;i<=n;i++){
		cin>>cnt;
		preOrder[i]=cnt;
	}
	temp=1;
	TreeNode *cnt_root=new TreeNode(preOrder[temp]);
	check_inOrder[0]=1;
	cnt_root=build(cnt_root);
	levelOrder(cnt_root);
}
signed main() {
	iloves;
	int test = 1;
	while (test--) {
		slove();
	}
}