027根據前序遍歷和中序遍歷,重建二叉樹(keep it up)

劍指offer中題目:http://ac.jobdu.com/problem.php?pid=1385

題目描述:

輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建出該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重複的數字。例如輸入前序遍歷序列{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
這個題用遞歸來重建二叉樹,具體代碼如下:

#include <stdio.h>
#include <stdlib.h>
 
#define SIZE 1001
 
typedef struct STreeNode
{
    int data;
    struct STreeNode *left;
    struct STreeNode *right;
}STreeNode;
 
int ArrPre[SIZE];
int ArrMid[SIZE];
int CanBuild;
 
int getMidPos(int vKey, int vStart, int vEnd)
{
    int i;
    for (i = vStart; i < vEnd; ++i)
    {
        if (ArrMid[i] == vKey) return i;
    }
    return -1;
}
 
void createNode(STreeNode **vNode, int vData)
{
    *vNode = (STreeNode *)malloc(sizeof(STreeNode));
    (*vNode)->data  = vData;
    (*vNode)->left  = NULL;
    (*vNode)->right = NULL;
}
 
void destroyNode(STreeNode **vNode)
{
    (*vNode)->left  = NULL;
    (*vNode)->right = NULL;
    free(*vNode);
}
 
void createBinaryTree(STreeNode **vNode, int vPreIndex, int MidIndex, int vLen)
{
    if (vLen == 0) return;
 
    createNode(vNode, ArrPre[vPreIndex]);
    int MidPos = getMidPos(ArrPre[vPreIndex], MidIndex, MidIndex + vLen);
 
    if(MidPos == -1)
    {
        CanBuild = 0;
        return;
    }
 
    createBinaryTree(&((*vNode)->left), vPreIndex+1, MidIndex, MidPos-MidIndex);
    createBinaryTree(&((*vNode)->right), vPreIndex+(MidPos-MidIndex)+1,MidPos+1, vLen-1-(MidPos-MidIndex));
}
 
void postTravelTree(STreeNode *vNode)
{
    if (vNode == NULL) return;
    postTravelTree(vNode->left);
    postTravelTree(vNode->right);
    printf("%d ", vNode->data);
}
 
void destroyBinaryTree(STreeNode **vNode)
{
    if (*vNode == NULL) return;
    destroyBinaryTree(&((*vNode)->left));
    destroyBinaryTree(&((*vNode)->right));
    destroyNode(vNode);
}
 
int main()
{
    int N, i;
    STreeNode *Root;
 
    while (scanf("%d", &N) != EOF)
    {
        for (i = 0; i < N; ++i) scanf("%d", &ArrPre[i]);
        for (i = 0; i < N; ++i) scanf("%d", &ArrMid[i]);
 
        CanBuild = 1;
        createBinaryTree(&Root, 0, 0, N);
        if (!CanBuild) 
        {
            printf("No\n");
        }
        else
        {
            postTravelTree(Root);
            printf("\n");
        }
 
        destroyBinaryTree(&Root);
    }
 
    return 0;
}
/**************************************************************
    Problem: 1385
    User: 
    Language: C
    Result: Accepted
    Time:0 ms
    Memory:920 kb
****************************************************************/


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章