Suppose that all the keys in a binary tree are distinct positive integers. Given the preorder and inorder traversal sequences, you are supposed to output the first number of the postorder traversal sequence of the corresponding binary tree.
Input Specification:
Each input file contains one test case. For each case, the first line gives a positive integer N (≤ 50,000), the total number of nodes in the binary tree. The second line gives the preorder sequence and the third line gives the inorder sequence. All the numbers in a line are separated by a space.
Output Specification:
For each test case, print in one line the first number of the postorder traversal sequence of the corresponding binary tree.
Sample Input:
7
1 2 3 4 5 6 7
2 3 1 5 4 7 6
Sample Output:
3
题意:给出一个n表示序列中元素的个数,下面给出两行序列,分别是一棵二叉树的前序遍历序列和中序遍历序列。现在让你输出后序遍历序列的第一个数字。
思路:还是老样子,不建树了,直接递归处理。在一个前序遍历中,第一个节点一定是根节点,这样我们在中序遍历中找到这个节点,那么他的左边就是左子树了,他的右边就是右子树了。现在问题来了,左子树的根节点呢?就是前序遍历找到的节点的下一个节点,而右子树的根节点呢,就是当前前序遍历节点的值加上左子树的值,这样就得到了右子树的根节点。然后递归处理就完了。
#include<bits/stdc++.h>
#pragma GCC optimize(2)
#define MAXN 5050
#define maxnode 1000005
#define sigma_size 26
#define md 12345678
#define INF 0x3f3f3f3f
using namespace std;
int n;
int pre_order[MAXN],in_order[MAXN];
bool first;
void post_order(int root,int start,int endd)
{
if(start>endd)
return ;
int i=start;
while(i<endd&&pre_order[root]!=in_order[i])
i++;
post_order(root+1,start,i-1);
post_order(root+i-start+1,i+1,endd);
if(first)
printf("%d\n",pre_order[root]),first=false;
}
int main()
{
scanf("%d",&n);
for(int i=0;i<n;i++)
scanf("%d",&pre_order[i]);
for(int i=0;i<n;i++)
scanf("%d",&in_order[i]);
first=true;
post_order(0,0,n-1);
return 0;
}