劍指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
****************************************************************/