PAT L2-011. 玩转二叉树

L2-011. 玩转二叉树

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

输入第一行给出一个正整数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;
struct node
{
	int num;
	node *lchild;
	node *rchild;
	node(node *Lchild = NULL, node *Rchild = NULL) { lchild = Lchild;rchild = Rchild; }
};
void Sort(node* &root, int Pre[],int Mid[],int start, int end, int &k)//k用来跟踪目前所比较的前序序列元素
{
	if ((start > end) || (end < 0))
	{
		k--;
		return;
	}
	int m = 0;
	while (Mid[m] != Pre[k])
		m++;
	if(root==NULL)
	root = new node;
	root->num = Mid[m];
	Sort(root->lchild, Pre, Mid, start, m - 1, ++k);
	Sort(root->rchild, Pre, Mid, m + 1, end,++k);
}
int main()
{
	int num, count=0;
	cin >> num;
	int *Num1 = new int[num];
	int *Num2 = new int[num];
	queue<node*> Print;
	for (int i = 0;i < num;i++)
		cin >> Num1[i];
	for (int i = 0;i < num;i++)
		cin >> Num2[i];
	node* root = new node;
	root->num = Num1[0];
	Sort(root, Num2, Num1, 0, num - 1, count);
	Print.push(root);
	while (!Print.empty())//通过队列来实现层次的输出，先入右子树即可实现左右孩子互换
	{
		node* current = new node;
		current = Print.front();
		if ((current->rchild) != NULL)
			Print.push(current->rchild);
		if ((current->lchild) != NULL)
			Print.push(current->lchild);
		Print.pop();
		if (Print.empty())
			cout << current->num;
		else
			cout << current->num << " ";
	}
	return 0;
}