LeetCode105. 从前序与中序遍历序列构造二叉树

题目分析:根据前序的第一个节点就可以确定二叉树的根节点,然后在中序序列中找到该节点。位于该节点左侧的是根节点的左子树,位于该节点右侧的是根节点的右子树。然后根据左右子树节点的数量可以在前序序列中知道各自子树的前序序列。假设左子树节点的数量为3,那么除了前序的第一个节点,往后数三个就是该左子树的节点的前序序列,相应地,后面的节点就是右子树节点的前序序列。

将preorder[pl.....pr]记作前序序列,将inorder[il.....ir]记作相应的中序序列,不断递归下去。

代码展示:

/**
 * 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:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        return constructTree(preorder,0,preorder.size()-1,inorder,0,inorder.size()-1);
    }
    
    TreeNode* constructTree(vector<int>& preorder,int pl,int pr,vector<int>& inorder,int il,int ir){
        if(pl>pr || il>ir)
            return NULL;
        TreeNode* root = new TreeNode(preorder[pl]);
        for(int k=il;k<=ir;k++){
            if(preorder[pl]==inorder[k]){
                root->left = constructTree(preorder,pl+1,pl+k-il,inorder,il,k-1);
                root->right = constructTree(preorder,pl+k-il+1,pr,inorder,k+1,ir);
            }
        }
        return root;
    }
};

猜你喜欢

转载自blog.csdn.net/jaster_wisdom/article/details/81003555