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
输出样例:
时间限制
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
思路:这道题和树的遍历基本上是一样的 只不过一个给的是后序和中序,一个给的是先序和中序。
通过先序找到根节点,然后在中序中找到根节点的位置,那么根节点前面的就是根节点的左子树,后面的就是根节点的右子树。分别求出来左右子树中节点的个数,然后在子树中找到根节点,再重复上一步的操作,直达全部为叶子节点。
如图:
红色的为二叉树各节点的位置
由于本题要求的是镜面反转之后二叉树的层次遍历的结果,那么我在入队的时候,先入右子树,再入左子树即可。
AC代码:
#include <iostream>
#include <queue>
#include <vector>
#define maxn 1000
using namespace std;
int a[maxn], b[maxn];
struct node
{
int l;
int r;
}p[maxn];
int built(int al, int ar, int bl, int br)
{
if(bl > br)
return 0;
int root, p1, p2;
root = a[al];
p1 = bl;
while(b[p1]!=root) p1++;
p2 = p1 - bl;
p[root].l = built(al + 1, al + p2, bl, p1 - 1);
p[root].r = built(al + p2 + 1, ar, p1 + 1, br);
return root;
}
void bfs(int x)
{
queue<int>q;
vector<int>v;
q.push(x);
while(!q.empty())
{
int n = q.front();
q.pop();
if(n==0)
break;
v.push_back(n);
if(p[n].r!=0) q.push(p[n].r); ///先入右子树
if(p[n].l!=0) q.push(p[n].l); ///后入左子树
}
int len = v.size();
for(int i = 0; i < len; i++)
{
if(i==len - 1)
{
cout << v[i] << endl;
}
else
{
cout << v[i] << " ";
}
}
return ;
}
int main()
{
int n;
cin >> n;
for(int i = 0; i < n; i++)
{
cin >> b[i]; ///中序
}
for(int i = 0; i < n; i++)
{
cin >> a[i]; ///先序
}
built(0, n - 1, 0, n - 1);
int root = a[0];
bfs(root);
return 0;
}