7-5 根据后序和中序遍历输出先序遍历 分数 25
本题要求根据给定的一棵二叉树的后序遍历和中序遍历结果,输出该树的先序遍历结果。
输入格式:
第一行给出正整数N(≤30),是树中结点的个数。随后两行,每行给出N个整数,分别对应后序遍历和中序遍历结果,数字间以空格分隔。题目保证输入正确对应一棵二叉树。
输出格式:
在一行中输出Preorder:
以及该树的先序遍历结果。数字间有1个空格,行末不得有多余空格。
输入样例:
7
2 3 1 5 7 6 4
1 2 3 4 5 6 7
输出样例:
Preorder: 4 1 3 2 6 5 7
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int n; // 节点总数
int post[N]; // 记录后序遍历
int mid[N]; // 记录中序遍历
struct node
{
int data;
node *l, *r;
};
node *build(int len, int *post, int *mid) //建树
{
node *root = new node;
if (!len)
return NULL; // 返回空节点
root->data = post[len - 1]; // 后序遍历的最后第一个值为根节点
int i;
for (i = 0; i < len; i++)
{
if (mid[i] == post[len - 1]) // 在中序遍历中找到根节点
break;
}
root->l = build(i, post, mid); // 构建左子树
root->r = build(len - i - 1, post + i, mid + i + 1); // 构建右子树
return root;
}
void preorder(node *root) //前序遍历
{
if (root)
{
cout << " " << root->data;
preorder(root->l);
preorder(root->r);
}
}
int main()
{
cin >> n;
for (int i = 0; i < n; i++)
cin >> post[i];
for (int i = 0; i < n; i++)
cin >> mid[i];
node *root = build(n, post, mid); // 建树
cout << "Preorder:";
preorder(root); //前序遍历
return 0;
}