L2-011. 玩转二叉树

最新推荐文章于 2024-04-18 14:56:15 发布

原创最新推荐文章于 2024-04-18 14:56:15 发布 · 215 阅读

0 ·

CC 4.0 BY-SA版权

PAT 同时被 2 个专栏收录

38 篇文章

订阅专栏

二叉树

4 篇文章

订阅专栏

本文介绍了一道编程题目，要求根据给定的中序和前序遍历构建二叉树，对其进行镜像操作后输出层序遍历的结果。文中提供了一个完整的C++实现代码，通过构建二叉树并利用队列完成遍历。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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 <stdio.h>
#include <math.h>
#include <string.h>
#include <algorithm>
#include <queue>
using namespace std;
#define LL long long
#define lt x<<1
#define rt x<<1|1
const int maxn = 1e4+111;
const int inf = 0x3f3f3f3f;
int a[maxn],b[maxn];
struct node
{
	int l,r,val;
}tree[maxn];
int cnt=0;
int build(int la,int ra,int lb,int rb)
{
	if(la>ra) return 0;
	int root = ++cnt;
	tree[root].val=b[lb];
	int i;
	for(i=la;i<=ra;i++){
		if(a[i]==b[lb]) break;
	}
	tree[root].l=build(la,i-1,lb+1,lb+i-la);
	tree[root].r=build(i+1,ra,lb+i-la+1,rb);
	return root;
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<=n;i++) scanf("%d",&b[i]);
	int root = build(1,n,1,n);
	queue<int> q;
	q.push(root);
	cnt=0;
	while(!q.empty()){
		int now=q.front();
		q.pop();
		cnt++;
		printf("%d",tree[now].val);
		if(tree[now].r!=0) q.push(tree[now].r);
		if(tree[now].l!=0) q.push(tree[now].l);
		if(cnt==n) printf("\n");
		else printf(" ");
	}

	return 0;
}