leetcode-Construct Binary Tree from Preorder and Inorder Traversal

来源:互联网 发布:java 7 并发编程指南 编辑:程序博客网 时间:2024/05/17 08:57

Given preorder and inorder traversal of a tree, construct the binary tree.

/** * Definition for binary tree * struct TreeNode { *     int val; *     TreeNode *left; *     TreeNode *right; *     TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */class Solution {public:    TreeNode *build(vector<int>&pre,int prestart,int preend,vector<int>&in,int instart,int inend)    {        if(preend<prestart||inend<instart||(preend-prestart)!=(inend-instart))            return NULL;        TreeNode *head = new TreeNode(pre[prestart]);        int rootpos;        for(int i=0;i<=inend-instart;i++)            if(in[i+instart]==pre[prestart])            {                rootpos = i;                break;            }                    head->left = build(pre,prestart+1,prestart+rootpos,in,instart,instart+rootpos-1);        head->right = build(pre,prestart+rootpos+1,preend,in,instart+rootpos+1,inend);        return head;    }    TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder)     {        int presize = preorder.size();        int insize   = inorder.size();        TreeNode *head = build(preorder,0,presize-1,inorder,0,insize-1);        return head;    }};


0 0
原创粉丝点击