剑指offer--重建二叉树

匿名 (未验证) 提交于 2019-12-03 00:37:01

题目描述:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

  • 必备知识点:前序遍历(根左右), 中序遍历(左根右),前序遍历第一个数1必是根节点,对应到中序遍历中根节点的位置,左边的数位于左子树,右边的数位于右子树;可以看出,1节点同时拥有左右子树;然后, 前序遍历中第二个数字2作为节点,对应到中序遍历中,左边为左子树,同时,右边没有数,所以2节点没有右子树。所以,可以通过递归实现二叉树重建。
TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin) { 		if (pre.empty() == NULL || vin.empty() == NULL) 			return NULL; 		size_t index = 0; 		for (; index < vin.size(); index++) 		{ 			if (vin[index] == pre[0]) 				break; 		} 		TreeNode* root = new TreeNode(pre[0]);  		vector<int> pre_left, vin_left; 		vector<int> pre_right, vin_right;  		for (size_t j = 0; j < index; j++) 		{ 			pre_left.push_back(pre[j + 1]); 			vin_left.push_back(vin[j]); 		} 		for (size_t j = index + 1; j < vin.size(); j++) 		{ 			pre_right.push_back(pre[j]); 			vin_right.push_back(vin[j]); 		}  		root->left = reConstructBinaryTree(pre_left, vin_left); 		root->right = reConstructBinaryTree(pre_right, vin_right);  		return root; 	}

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!