【劍指offer】題目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”。

樣例輸入：

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;
}