题目2：重建二叉树

题目：http://ac.jobdu.com/problem.php?cid=1039&pid=1

题目描述：

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并输出它的后序遍历序列。

输入：

输入可能包含多个测试样例，对于每个测试案例，

输入的第一行为一个整数n(1<=n<=1000)：代表二叉树的节点个数。

输入的第二行包括n个整数(其中每个元素a的范围为(1<=a<=1000))：代表二叉树的前序遍历序列。

输入的第三行包括n个整数(其中每个元素a的范围为(1<=a<=1000))：代表二叉树的中序遍历序列。

输出：

对应每个测试案例，输出一行：

如果题目中所给的前序和中序遍历序列能构成一棵二叉树，则输出n个整数，代表二叉树的后序遍历序列，每个元素后面都有空格。

如果题目中所给的前序和中序遍历序列不能构成一棵二叉树，则输出”No”。

样例输入：

8

1 2 4 7 3 5 6 8

4 7 2 1 5 3 8 6

8

1 2 4 7 3 5 6 8

4 1 2 7 5 3 8 6

样例输出：

7 4 2 5 8 6 3 1

 No

本题主要是在给出前序遍历和中序遍历的情况下构造出后序遍历，如果不存在就输出No!做这题目时，最主要的就是前序遍历的第一个元素一定是根结点!从而在中序遍历中找到根元素，根元素左边的在根结点的左子树，根元素的右边的位于根结点的右子树，从面根据后序遍历的定义，构造出后序遍历，如果出现不符，则不能构造出，直接输出 No!
以下是AC代码：
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <cctype>
#include <cstdlib>
#include <string.h>
#include <algorithm>
#include <cmath>
using namespace std;


#define N 1010


int flag;//用于判断是否存在


int pos(int p[],int t,int n)
{
    int i;
    for(i = 0;i < n;i++)
          if(p[i] == t)
                  return i ;
    flag = 0; //不存在
    return -1;
}
void build(int n,int s1[],int s2[],int s[])
{
     if(!flag)
              return;
     if(n <= 0)
          return;
     int p = pos(s2,s1[0],n);//找根结点
     build(p,s1 + 1 ,s2,s);//构造左子树
     build(n - p - 1,s1  +  p  + 1,s2  +  p  +  1,s  +  p);//构造右子树
     s[n - 1] = s1[0];
}
int main()
{
    int p[N];
    int q[N];
    int ans[N];
    int n,i;
    while(scanf("%d",&n) == 1)
    {
                         flag = 1;
                         for(i = 0;i < n;i++)
                               scanf("%d",&p[i]);
                         for(i = 0;i < n;i++)
                               scanf("%d",&q[i]);
                         build(n,p,q,ans);
                         if(flag)
                         {
                             for(i = 0;i < n ;i++)
                                   printf("%d ",ans[i]);
                             printf("\n");
                         }
                         else
                             printf("No\n");
    }
    return 0;
}