数据结构作业——order（二叉树遍历）

order

Description

给出一棵二叉树的中序遍历和每个节点的父节点，求这棵二叉树的先序和后 序遍历。

Input

输入第一行为一个正整数 n 表示二叉树的节点数目， 节点编号从 1 到 n，其中 1 为根节点。
第 2 行有 n 个数字， 第 i 个数字表示 i 的父亲节点。（ 1 的父亲节点为 0， 表示无）
第 3 行为中序遍历。
30%的数据： n<=20；
60%的数据： n<=1000；
100%的数据： n<=10000；

Output

输出 2 行， 第一行为先序遍历，第二行为后序遍历。

Sample Input

10
0 7 2 2 9 1 8 1 6 8
9 5 6 1 10 8 7 3 2 4

Sample Output

1 6 9 5 8 10 7 2 3 4
5 9 6 10 3 4 2 7 8 1

思路

存下每个节点的左右儿子，然后根据中序遍历判断谁是左儿子，谁是右儿子，建完树，跑一下先序遍历和后序遍历

 

+ View Code?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75

#include<iostream>  #include<cstdio>  #include<cstring>  using namespace std;  const int maxn = 10005;  int lson[maxn],rson[maxn],pos[maxn];  bool first = true;      void PreOrder(int x)  {      if (x != 1) printf(" ");      printf("%d",x);      if (lson[x])    PreOrder(lson[x]);      if (rson[x])    PreOrder(rson[x]);  }      void PostOrder(int x)  {      if (lson[x])    PostOrder(lson[x]);      if (rson[x])    PostOrder(rson[x]);      first?printf("%d",x):printf(" %d",x);      first = false;  }      int main()  {      //freopen("input.txt","r",stdin);      //freopen("output.txt","w",stdout);      int N,i,tmp;      memset(lson,0,sizeof(lson));      memset(rson,0,sizeof(rson));      memset(pos,0,sizeof(pos));      scanf("%d",&N);      for (i = 1;i <= N;i++)      {          scanf("%d",&tmp);          if (tmp == 0)   continue;          if (!lson[tmp]) lson[tmp] = i;          else    rson[tmp] = i;        }      for (i = 1;i <= N;i++)      {          scanf("%d",&tmp);          pos[tmp] = i;      }      for (i = 1;i <= N;i++)      {          if (lson[i] && rson[i])          {              if (pos[rson[i]] < pos[lson[i]])              {                  swap(rson[i],lson[i]);              }          }          else if (!lson[i] && rson[i])          {              if (pos[rson[i]] < pos[i])              {                  lson[i] = rson[i];                  rson[i] = 0;              }          }          else if (lson[i] && !rson[i])          {              if (pos[lson[i]] > pos[i])              {                  rson[i] = lson[i];                  lson[i] = 0;              }          }      }      PreOrder(1);printf("\n");      PostOrder(1);printf("\n");      return 0;  }