本题要求根据给定的一棵二叉树的先序遍历和中序遍历结果,输出该树的后序遍历结果。
输入格式:
第一行给出正整数N(≤30),是树中结点的个数。随后两行,每行给出N个整数,分别对应先序遍历和中序遍历结果,数字间以空格分隔。题目保证输入正确对应一棵二叉树。
输出格式:
在一行中输出Postorder:
以及该树的后序遍历结果。数字间有1个空格,行末不得有多余空格。
输入样例:
7
4 1 3 2 6 5 7
1 2 3 4 5 6 7
输出样例:
Postorder: 2 3 1 5 7 6 4
输入样例:
5
3 9 20 15 7
9 3 15 20 7
输出样例:
Postorder:9 15 7 20 3
思路:我们可以通过先序遍历快速找到根,再带入中序遍历中找出左右子树,分别递归左右子树将每次找到的根输出
#include<iostream>
using namespace std;
void postOrder(int n,int* preorder,int* inorder) {
if (n < 0) return; //返回上一级函数
int i = 0;
for (i; i <= n; i++) {
if (preorder[0] == inorder[i]) break;
}
postOrder(i-1,preorder+1,inorder);//对应左子树
postOrder(n-i-1,preorder+i+1,inorder+i+i);//对应右子树
printf("%d ", preorder[0]);//最后输出根节点
return;
}
int main() {
int n;
cin >> n;
int* preorder = (int*)malloc(sizeof(int) * n);
int* inorder = (int*)malloc(sizeof(int) * n);
for (int i = 0; i < n; i++) {
cin >> preorder[i];
}
for (int i = 0; i < n; i++) {
cin >> inorder[i];
}
cout << "Postorder:";
postOrder(n-1, preorder, inorder);
return 0;
}