By

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

题目解析

其实这题的思路还是挺清晰的，方法就是递归。前序遍历序列的第一个节点就是根节点，中序遍历序列的这个节点的左边就是它的左子树，右边就是它的右子树，前序遍历中左子树在右子树的前面。再把左右子树分别的前序遍历序列和中序遍历序列拿去递归～递归～递归～代码如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    struct TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> in) {
    int i,j,k;
    vector<int>pre_left,pre_right,in_left,in_right;
    struct TreeNode *head;
    head=new TreeNode(pre[0]);
 
    //用root_index记录根结点在中序遍历的下标
    int root_index;
    for(i=0;i<in.size();i++){
        if(pre[0]==in[i]){
            root_index=i;
        }
    }
 
    //把相应的左子树和右字树放到vector中递归
    for(i=0;i<root_index;i++){
        pre_left.push_back(pre[i+1]);
        in_left.push_back(in[i]);
    }
    for(i=root_index;i<in.size()-1;i++){
        pre_right.push_back(pre[i+1]);
        in_right.push_back(in[i+1]);
    }
 
    //递归
    if(pre_left.size()!=0)
        head->left=reConstructBinaryTree(pre_left,in_left);
    else
        head->left=NULL;
    if(pre_right.size()!=0)
        head->right=reConstructBinaryTree(pre_right,in_right);
    else
        head->right=NULL;
 
    return head;
    }
};