7-3 树的遍历 (25 分)

题目

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数N（≤30），是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。

输出格式：

在一行中输出该树的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7
结尾无空行

输出样例：

4 1 6 3 5 7 2
结尾无空行

思路

根据后序遍历的特点可知，后序遍历的序列的最后一个结点就是树的根节点。再从中序遍历中找到该根节点的位置，那么此位置的左部分结点必为左子树，右半部分必为右子树，得到左子树的结点共有 index个， 右子树结点共有 N-index-1个。

再次调用函数时，后序遍历序列中从第0个结点开始（左子树）和从第index个结点开始（右子树）；中序遍历序列从第0开始（左子树）和从第index+1开始（右子树）。

层次遍历的输出借鉴了队列的知识，代码中使用了数组去模拟队列.

注意递归的终止条件: 在中序遍历中找不到根节点，此时的index=0

数据输入时，注意换行符

代码

#include <stdio.h>
#include <stdlib.h>
typedef struct TNode *Tree;

struct TNode
{
    int data;
    struct TNode *Left;
    struct TNode *Right;
};
Tree queue[100];

Tree BuildTree(int inorder[], int postorder[], int n)
{
    if(!n) return NULL;
    Tree T = (Tree) malloc(sizeof(struct TNode));
    T->data = postorder[n-1]; //后序遍历的最后一个节点就是树的根节点
    T->Left = T->Right = NULL;

    //从中序遍历中找到根节点的位置
    int index;
    for(index=0; index<n; index++)
    {
        if(inorder[index] == postorder[n-1]) break;
    }
    T->Left = BuildTree(inorder, postorder, index);
    T->Right = BuildTree(inorder+index+1, postorder+index, n-index-1);
    return T;
}

void LevelOrder(Tree T)
{
    if(T)
    {
        int left = 0, right = 0;
        queue[right++] = T;
        while (left < right)
        {
            Tree bt = queue[left++];
            if(bt == T) printf("%d", bt->data); //为根节点时，无空格输出
            else printf(" %d", bt->data);

            if(bt->Left) queue[right++] = bt->Left;
            if(bt->Right) queue[right++] = bt->Right;
        }
    }
}

int main()
{
    int inorder[100]; //中序遍历
    int postorder[100]; //后序遍历
    int n;
    scanf("%d", &n);
    for(int i=0; i<n; i++)
        scanf("%d", &postorder[i]);
    for(int i=0; i<n; i++)
        scanf("%d", &inorder[i]);
    Tree T = BuildTree(inorder, postorder, n);
    LevelOrder(T);
    return 0;
}