LeetCode:106.从中序与后序遍历序列构造二叉树

难度:中等

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

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

例如,给出

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

返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7

C++:递归算法

类似于前序与中序遍历构造二叉树

要点:
	1、中序遍历序列中最后一个节点为根节点;
	2、中序遍历序列中根节点的左侧为左子树的节点,右侧为右子树的节点
思路:
	1、构造一个函数,该函数的参数为(前序遍历序列,(子)中序序列的起始位置,(子)中序序列的终止位置,后序遍历序列,(子)后序序列的起始位置,(子)后序序列的终止位置);
	2、根据(子)后序序列的最后一个位置的值创建根节点;
	3、查找根节点的值在中序遍历序列中的序号(中根序号);
	4、依据中根序号-(子)中序序列的起始位置,得到左子树有多少个节点(左子树节点数);
	5、递归构建根节点的左孩子,此时(子)中序序列的起始位置不变,(子)中序序列的终止位置=中根序号-1,(子)后序序列的起始位置不变,(子)后序序列的终止位置=(子)后序序列的起始位置+左子树节点数-1;
	6、递归构建根节点的右孩子,此时(子)中序序列的起始位置=中根序号+1,(子)中序序列的终止位置不变,(子)后序序列的起始位置=(子)后序序列的起始位置+左子树节点数,(子)后序序列的终止位置=(子)后序序列的终止位置-1;
	7、返回根节点

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector& inorder, vector& postorder) {
        return buildFromInAndPost(inorder, 0, inorder.size()-1, postorder, 0, postorder.size()-1);
    }

    TreeNode* buildFromInAndPost(vector& inorder, int inLeft, int inRight, vector& postorder, int postLeft, int postRight){
        if(inLeft > inRight || postLeft > postRight) return nullptr;
        TreeNode* root = new TreeNode(postorder[postRight]);//将后序遍历的最后一个值创建根节点
        int inRoot = inLeft;
        while(inRoot <= inRight && root->val != inorder[inRoot])//查找中序遍历序列中的根节点
            inRoot++;
        int leftCount = inRoot - inLeft;//判断左子树有多少个节点
        root->left = buildFromInAndPost(inorder, inLeft, inRoot-1, postorder, postLeft, postLeft+leftCount-1);
        root->right = buildFromInAndPost(inorder, inRoot+1, inRight, postorder, postLeft+leftCount, postRight-1);
        return root;
    }
};

你可能感兴趣的:(算法)