【LeetCode】—— 构建二叉树LeetCode105题

根据一棵树的前序遍历与中序遍历构造二叉树。 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;
     *     struct TreeNode *left;
     *     struct TreeNode *right;
     * };
     */
     //为方便递归创建的子函数
struct TreeNode* _buildTree(int* preorder,int* ppreindex,int* inorder,int inbegin,int inend)
    {
        //若区间中没有元素，返回空
        if(inbegin > inend)
            return NULL;
        struct TreeNode* root = (struct TreeNode*)malloc(sizeof(struct TreeNode));
        root->val = preorder[*ppreindex];
        ++(*ppreindex);
    //若区间中只有一个元素，可直接将该元素当做根节点构建二叉树，左右子树均为空，并返回根
        if(inbegin == inend)
        {
            root->left = NULL;
            root->right = NULL;
            return root;
        }
        
        //若区间不止一个元素，划分子问题解决
        //根，左子树，右子树
        //根据前序遍历找到的根节点在中序遍历中找到对应的根节点将左右子树的区间划分
        int rootindex = inbegin; //定义中序遍历中根节点的下标
        while(rootindex <= inend)
        {
            if(inorder[rootindex] == root->val)
                break;
            else
                ++rootindex;
        }
        assert(rootindex <= inend);
        
        //左子树所在区间为[inbegin,rootindex-1]，构造左子树
        root->left = _buildTree(preorder,ppreindex,inorder,inbegin,rootindex-1);
        //右子树所在区间为[rootindex+1,inend]，构造右子树
         root->right = _buildTree(preorder,ppreindex,inorder,rootindex+1,inend);
        
        return root;  //构造成功返回根节点
    }
    struct TreeNode* buildTree(int* preorder, int preorderSize, int* inorder, int inorderSize) {
        int preindex = 0;
       return _buildTree(preorder,&preindex,inorder,0,inorderSize-1);
    }