LeetCode 面试题07. 重建二叉树

面试题07. 重建二叉树

难度中等

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

 

例如，给出

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

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

 

限制：

0 <= 节点个数 <= 5000

思路：在先序遍历的数据中确定根节点，在中序遍历的数据中确定左右子树。利用后序遍历，中序遍历数据恢复二叉树方法相似，只需做一些细微的改动。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */


struct TreeNode* buildTree(int* preorder, int preorderSize, int* inorder, int inorderSize){
    struct TreeNode *root;
    int i;
    if(preorderSize<=0){
        return NULL;
    }
    root=(struct TreeNode*)malloc(sizeof(struct TreeNode));
    root->val=*preorder;
    for(i=0;i<inorderSize;i++){
        if(inorder[i]==preorder[0]){
            break;
        }
    }
    root->left=buildTree(preorder+1,i,inorder,i);
    root->right=buildTree(preorder+i+1,preorderSize-i-1,inorder+i+1,inorderSize-i-1);
    return root;
}