Construct Binary Tree from Inorder and Postorder Traversal 从中序与后序遍历序列构造二叉树

根据一棵树的中序遍历与后序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如,给出

中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]

返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7

思路和前一道题一样点我,通过根节点把中序序列和后序序列分成三部分,然后依次递归调用自身。

TreeNode* buildTreeCore(vector<int>& inorder, vector<int>& postorder, int in_start, int in_end, int post_start, int post_end) {
	if (in_start > in_end || post_start > post_end) {
		return nullptr;
	}
	if (in_start == in_end && post_start == post_end) {
		return new TreeNode(inorder[in_start]);
	}
	TreeNode *node = new TreeNode(postorder[post_end]);
	int i = inorder.size()-1;
	for (; i >=in_start; i--) {
		if (inorder[i] == postorder[post_end]) {
			break;
		}
	}
	//int in_mid = getMid(inorder, in_start, in_end, preorder[pre_start]);
	//int pre_mid = ((in_mid - 1 )>= in_start )? getMid(preorder, in_start, in_end, inorder[in_mid - 1]) : -1;
	node->left = buildTreeCore(inorder, postorder, in_start, i-1, post_start, post_end-(in_end - i)-1);
	node->right = buildTreeCore(inorder, postorder, i+1, in_end, post_end-(in_end - i), post_end-1);
	return node;
}


TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
	if (inorder.empty() || postorder.empty()) {
		return nullptr;
	}
	return buildTreeCore(inorder, postorder, 0, inorder.size()-1, 0, postorder.size()-1);
}





猜你喜欢

转载自blog.csdn.net/qq_26410101/article/details/80545753