Lintcode---中序遍历和后序遍历树构造二叉树

根据中序遍历和后序遍历树构造二叉树

 注意事项

你可以假设树中不存在相同数值的节点

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出树的中序遍历： [1,2,3] 和后序遍历： [1,3,2]

返回如下的树：

  2

 /  \

1    3

 

 

思路：与中序遍历和前序遍历构造二叉树的过程类似。只不过对于后序遍历来说。根节点是最后一个被访问的节点；
    
     或者，可以先将先序遍历序列逆序，然后过程就与中序遍历和前序遍历构造二叉树的过程完全一样了；
          
     

 

     注意：最后递归的时候，一定不要把中序遍历的序列和后续遍历的序列弄反了。血的教训，不要直接复制代码，顺着思路过一遍.

 

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
 

class Solution {
    /**
     *@param inorder : A list of integers that inorder traversal of a tree
     *@param postorder : A list of integers that postorder traversal of a tree
     *@return : Root of a tree
     */
    /*
    
    思路：与中序遍历和前序遍历构造二叉树的过程类似。只不过对于后序遍历来说。根节点是最后一个被访问的节点；
    
          或者，可以先将先序遍历序列逆序，然后过程就与中序遍历和前序遍历构造二叉树的过程完全一样了；
          
          最后递归的时候，一定不要把中序遍历的序列和后续遍历的序列弄反了。血的教训，不要直接复制代码，顺着思路过一遍
    
    */
public:
    TreeNode *buildTree(vector<int> &inorder, vector<int> &postorder) {
        // write your code here

        if(postorder.size()==0||inorder.size()==0){
            return NULL;
        }
        
        if(postorder.size()!=inorder.size()){
            return NULL;
        }
        
        vector<int> inorder_l,inorder_r,postorder_l,postorder_r;
        int root_index=-1;
        int len = postorder.size();

            
        TreeNode* root=new TreeNode(postorder[len-1]);
        

        //  在中序队列中找出根节点位置
        for(int i=0;i<len;i++){
            if(postorder[len-1]==inorder[i]){
                root_index=i;
                break;
            }
        }
        
        //  左右子树的后序、中序队列
        for(int i=0; i<root_index; i++) {
            postorder_l.push_back(postorder[i]);
            inorder_l.push_back(inorder[i]);
        }
        
        for(int i=root_index+1; i<inorder.size(); i++) {
            postorder_r.push_back(postorder[i-1]);
            inorder_r.push_back(inorder[i]);
        }

        
        root->left=buildTree(inorder_l, postorder_l);
        root->right=buildTree(inorder_r, postorder_r);
        
        return root;   
    }
};