leetcode105:從前序與中序遍歷序列構造二叉樹

---

---

一、題目

網址：從前序與中序遍歷序列構造二叉樹。

難度中等。

根據一棵樹的前序遍歷與中序遍歷構造二叉樹。

注意:
你可以假設樹中沒有重複的元素。

例如，給出

前序遍歷 preorder = [3,9,20,15,7]
中序遍歷 inorder = [9,3,15,20,7]

返回如下的二叉樹：

    3
   / \
  9  20
    /  \
   15   7

二、程序

/**
 * 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) {
        if(preorder.size()==0) return NULL;

        TreeNode *r=new TreeNode(preorder[0]);
        int find=-1;
        for(int i=0;i<inorder.size();i++)
        {
            if(r->val==inorder[i])
            {
                find=i;
                break;
            }
        }
        if(find==-1) return NULL;
        vector<int> lf_pre;
        vector<int> lf_ino;
        vector<int> ri_pre;
        vector<int> ri_ino;

        for(int i=0;i<find;i++)
        {
            lf_pre.push_back(preorder[i+1]);
            lf_ino.push_back(inorder[i]);
        } 
        r->left=buildTree(lf_pre,lf_ino);

        for(int i=find+1;i<preorder.size();i++)
        {
            ri_pre.push_back(preorder[i]);
            ri_ino.push_back(inorder[i]);

        } 
        r->right=buildTree(ri_pre,ri_ino);

        return r;

    }
};

結果：