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

来源:互联网 发布:java工程师好找工作吗 编辑:程序博客网 时间:2024/06/07 00:16

题目描述:
根据中序遍历和后序遍历树构造二叉树

注意事项:
你可以假设树中不存在相同数值的节点

样例:
给出中序遍历:[1,2,3]和后序遍历:[1,3,2]. 返回如下的树:

这里写图片描述

思路跟上一篇一样,唯一的不同是这里是从后序遍历找出根节点,即后序遍历数组的最后一个是根节点。上一篇是从先序遍历找出根节点,即先序遍历数组的第一个是根节点。

代码详解:

class Solution {public:    /*     * @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     */    TreeNode * buildTree(vector<int> &inorder, vector<int> &postorder) {        // write your code here        TreeNode *root;        if(inorder.size()==0)        {            root=NULL;            return root;        }        int rootNode=postorder[postorder.size()-1];        int index=findpos(inorder,rootNode);        vector<int>left_inorder;        vector<int>right_inorder;        for(int i=0;i<index;i++){            left_inorder.push_back(inorder[i]);        }        for(int i=index+1;i<inorder.size();i++){            right_inorder.push_back(inorder[i]);        }        vector<int>left_postorder;        vector<int>right_postorder;        for(int i=0;i<index;i++)        {            left_postorder.push_back(postorder[i]);        }        for(int i=index;i<postorder.size()-1;i++)        {            right_postorder.push_back(postorder[i]);        }        root=new TreeNode();        if(root!=NULL)        {            root->val=rootNode;            root->left=buildTree(left_inorder,left_postorder);            root->right=buildTree(right_inorder,right_postorder);        }        return root;    }    int findpos(vector<int>inorder,int m)    {        for(int i=0;i<inorder.size();i++)        {            if(inorder[i]==m)            {                return i;            }        }        return 0;    }};
阅读全文
0 0