LeetCode 从中序与后序遍历序列构造二叉树(递归+图解)

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

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

例如,给出

中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7

请先翻阅 LeetCode 从前序与中序遍历序列构造二叉树
此题与上一题基本一样的解法。

中序遍历:先左子树,后根节点,再右子树
后序遍历:先左子树,后右子树,再根节点
LeetCode 从中序与后序遍历序列构造二叉树(递归+图解)_第1张图片

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
	int orderSize;
	vector inorder;//类的属性,作用类似全局遍历
	vector postorder;
	//inorder [inorderBegin, inorderEnd], postorder [postorderBegin, postorderEnd] 构造成一棵树
	TreeNode* myBuildTree(int inorderBegin, int inorderEnd, int postorderBegin, int postorderEnd) {
		TreeNode *root = NULL;
		if (inorderBegin == inorderEnd) {
			root = new TreeNode(inorder[inorderBegin]);
		}
		else if (inorderBegin < inorderEnd) {
			root = new TreeNode(postorder[postorderEnd]);
			int tempValue = postorder[postorderEnd];//根节点的值,因为postorder后序遍历最后访问的根节点
			int rootIndex = inorderBegin;//根节点在inorder的下标
			while (rootIndex < inorderEnd && inorder[rootIndex] != tempValue) {//寻找根节点在inorder的下标
				++rootIndex;
			}
			int leftCnt = rootIndex - inorderBegin;//左子树的节点个数
			root->left = myBuildTree(inorderBegin, rootIndex -1, postorderBegin, postorderBegin + leftCnt - 1);//构建左子树
			root->right = myBuildTree(rootIndex + 1, inorderEnd, postorderBegin + leftCnt, postorderEnd - 1);//构建右子树
		}
		return root;
	}
	TreeNode* buildTree(vector& inorder, vector& postorder) {
		TreeNode* root = NULL;
		orderSize = inorder.size();
		int postorderSize = inorder.size();
		if (orderSize == NULL || orderSize != postorderSize) {
			return NULL;
		}
		this->inorder = inorder;//复制
		this->postorder = postorder;
		root = myBuildTree(0, orderSize - 1, 0, orderSize - 1);//构造
		return root;
	}
};

LeetCode 从中序与后序遍历序列构造二叉树(递归+图解)_第2张图片
作图的时间确实不少,希望没有理解的道友,看完我的博客能恍然大悟。

你可能感兴趣的:(LeetCode)