【剑指offer】重建二叉树

标签：c++   面试   

struct BinaryTreeNode{
	int m_nValue;
	BinaryTreeNode *m_pLeft;
	BinaryTreeNode *m_pRight;
};

        创建根节点。前序遍历的第一个数据就是根节点的数据，在中序遍历中找到根节点的位置，可分别得知左子树和右子树的前序和中序遍历序列，递归重建左右子树。

BinaryTreeNode *ReBuiltBiTree(int *preOrder,int *inOrder, int treeNum)
{
	if (preOrder==NULL||inOrder==NULL||treeNum<=0)
	{
		return NULL;
	}
	//前序遍历的第一个数字是根节点
	BinaryTreeNode *root=new BinaryTreeNode;
	root->m_nValue=preOrder[0];
	root->m_pRight=root->m_pLeft=NULL;
	//在中序遍历中找到根结点，则根结点左边结点是左子树，右边结点是右子树
	int rootorder=-1;
	for (int i=0; i<treeNum; i++)
	{
		if (root->m_nValue==inOrder[i])
		{
			rootorder=i;
			break;
		}
	}
	if (rootorder==-1)
	{
		printf("Invalid Input");
	}
	//重建左子树
	int treeNumleft=rootorder;
	int *preOrderleft=preOrder+1;
	int *inOrderleft=inOrder;
	root->m_pLeft=ReBuiltBiTree(preOrderleft,inOrderleft,treeNumleft);
	//重建右子树
	int treeNumRight=treeNum-treeNumleft-1;
	int *preOrderRight=preOrder+treeNumleft+1;
	int *inOrderRight=inOrder+treeNumleft+1;
	root->m_pRight=ReBuiltBiTree(preOrderRight,inOrderRight,treeNumRight);
	return root;
}

【剑指offer】重建二叉树

标签：c++   面试   

原文地址：http://blog.csdn.net/lsh_2013/article/details/45568233