根据后序和中序遍历输出先序遍历
本题要求根据给定的一棵二叉树的后序遍历和中序遍历结果,输出该树的先序遍历结果。
输入格式:
第一行给出正整数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
代码:
#include <algorithm>
#include <iostream>
#include<string.h>
#include<stdio.h>
#include <stack>
#include <map>
using namespace std;
int s1[40], s2[40];
struct node {
int data;
struct node *lchild, *rchild;
};
node* fun(int n1, int n2, int m1, int m2) {
node *head = new node();
head->data = s1[n2];
head->lchild = NULL;
head->rchild = NULL;
int i;
for (i = m1; i <= m2; i++) {
if (s2[i] == s1[n2]) {
if (i != m1) {
head->lchild = fun(n1, n1 + i - m1 - 1, m1, i - 1);
}
if (i != m2) {
head->rchild = fun(n2 - m2 + i, n2 - 1, i + 1, m2);
}
}
}
return head;
}
void putout(node *now) {
if (now == NULL) {
return;
}
cout << " " << (*now).data;
putout(now->lchild);
putout(now->rchild);
}
int main() {
int n;
cin >> n;
int i;
for (i = 0; i < n; i++) {
cin >> s1[i];
}
for (i = 0; i < n; i++) {
cin >> s2[i];
}
node *head;
head = fun(0, n - 1, 0, n - 1);
cout << "Preorder:";
putout(head);
cout << endl;
return 0;
}