您的位置:首页 > Web前端

【剑指offer】题目1385:重建二叉树

2014-04-04 13:04 429 查看
题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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”。

样例输入:
81 2 4 7 3 5 6 84 7 2 1 5 3 8 681 2 4 7 3 5 6 84 1 2 7 5 3 8 6


样例输出:
7 4 2 5 8 6 3 1 No


这代码有Bug,只过了一个样例,谁能帮我找出来?


#include <iostream>
#include <string>
#include <cstdio>

using namespace std;
int preorder[1001];
int inorder[1001];
bool isvalid = true;

struct TreeNode{
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int value){val = value; left = NULL; right = NULL;}
};

TreeNode *Construct(int *preorder, int *endpreorder, int *inorder, int *endinorder)
{
int rootvalue = preorder[0];
TreeNode *root = new TreeNode(rootvalue);
if (preorder == endpreorder)
{
if ( inorder == endinorder && *preorder == *inorder)
{
return root;
}
else {
isvalid = false;
return NULL;
}
}
int *rootinorder = inorder;
while(rootinorder <= endinorder  && *rootinorder != rootvalue)
rootinorder++;
if (rootinorder > endinorder)
{
isvalid = false;
return NULL;
}
int leftlength = rootinorder - inorder;
int rightlength = endinorder - rootinorder;
if (leftlength > 0)
{
root->left = Construct(preorder + 1, preorder + leftlength, inorder, rootinorder-1);
}
if (rightlength > 0)
{
root->right  = Construct(preorder+leftlength+1, endpreorder, rootinorder+1, endinorder);
}
return root;
}

TreeNode *ConstructTree(int *preorder, int *inorder, int length)
{
if (preorder == NULL || inorder == NULL || length <= 0)
{
return NULL;
}
return Construct(preorder, preorder+length-1, inorder, inorder+length-1);
}

void posttraversal(TreeNode *root)
{
if (root->left)
{
posttraversal(root->left);
}
if (root->right)
{
posttraversal(root->right);
}
printf("%d ",root->val);
}

void destroyTree(TreeNode *root)
{
if(root == NULL)
return;
if (root->left)
{
destroyTree(root->left);
}
if (root->right)
{
destroyTree(root->right);
}
delete root;
}

int main()
{

int n;
while(cin>>n)
{
for (int i = 0; i < n; i++)
{
scanf("%d", &preorder[i]);
}
for (int i = 0; i < n; i++)
{
scanf("%d", &inorder[i]);
}
TreeNode *root = ConstructTree(preorder, inorder, n);
if (root == NULL || !isvalid)
{
printf("No");
}
else posttraversal(root);
printf("\n");
destroyTree(root);
}
//system("pause");
return 0;
}


[/code]
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: