7-1 根据后序和中序遍历输出先序遍历 (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
结尾无空行
CODE:
using namespace std;
#include <algorithm>
#include<cstring>
#include <iostream>
#include<string>
#include<stdio.h>
#include<stdlib.h>
#include<time.h>
#include<string.h>
#include<iomanip>
#include<sstream>
#include<vector>;
#include<deque>
#include <set>
#include <unordered_set>
#include <map>
#include<stack>
#include<queue>
#include<list>
#define element int
struct TreeNode {
element val;
TreeNode* left;
TreeNode* right;
TreeNode() : val(0), left(nullptr), right(nullptr) {}
TreeNode(element x) : val(x), left(nullptr), right(nullptr) {}
TreeNode(element x, TreeNode* left, TreeNode* right) : val(x), left(left), right(right) {}
};
map<element, int>m;
TreeNode* building(const vector<element>& ior, const vector<element>& fin, int ior_l, int ior_r, int fin_l, int fin_r)
{
if (fin_l > fin_r)return NULL;
int fin_root = fin_r;//根节点在后序遍历数组中的下标
int ior_root = m[fin[fin_root]];//根节点在中序遍历数组中的下标
int L_size = ior_root - ior_l;//根节点左子树的大小
TreeNode* root = new TreeNode(fin[fin_root]);//创建根节点
root->left = building(ior, fin, ior_l, ior_root - 1, fin_l, fin_l + L_size - 1);
root->right = building(ior, fin, ior_root + 1, ior_r, fin_l + L_size, fin_root - 1);
return root;
}
vector<element>T;
void travel(TreeNode* root)
{
if (root)
{
T.push_back(root->val);
travel(root->left); travel(root->right);
}
}
int main()
{
int n, i; cin >> n;
element t;
vector<element>ior; vector<element>fin;
for (i = 0; i < n; i++)
{
cin >> t;//输入后序遍历数组
fin.push_back(t);
}
for (i = 0; i < n; i++)
{
cin >> t;
ior.push_back(t);//输入中序遍历数组
}
for (int i = 0; i < ior.size(); ++i) m[ior[i]] = i;
TreeNode* root = building(ior, fin, 0, ior.size() - 1, 0, fin.size() - 1);//构建二叉树
travel(root);//前序遍历
cout << "Preorder: ";
for (int i = 0; i < T.size(); i++)
{
if (i + 1 < T.size())cout << T[i] << " ";
else cout << T[i] << endl;
}
}