【每日打卡】Day22:重建二叉树 C++实现

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/qq_40184885/article/details/90141523

输入一棵二叉树前序遍历和中序遍历的结果,请重建该二叉树。

注意:

  • 二叉树中每个节点的值都互不相同;
  • 输入的前序遍历和中序遍历一定合法;

样例

给定:
前序遍历是:[3, 9, 20, 15, 7]
中序遍历是:[9, 3, 15, 20, 7]

返回:[3, 9, 20, null, null, 15, 7, null, null, null, null]
返回的二叉树如下所示:
    3
   / \
  9  20
    /  \
   15   7

 

 

代码:

class Solution {
public:
    map<int,int> loc;
    TreeNode* dfs(vector<int>& pre, vector<int>& in, int pstart, int pend, int istart, int iend){
        if(pstart > pend)
            return NULL;
        //k表示左子树节点的数目    
        int k = loc[pre[pstart]] - istart;
        //建立根节点
        TreeNode* root = new TreeNode(pre[pstart]);
        root->left = dfs(pre,in,pstart+1,pstart+k,istart,istart+k-1);
        root->right = dfs(pre,in,pstart+k+1,pend,istart+k+1,iend);
        
    }
    
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int size = preorder.size();
        for(int i = 0; i < size; i ++)
            loc[inorder[i]] = i;
        return dfs(preorder,inorder,0,size-1,0,size-1);    
        
    }
};

 

展开阅读全文

没有更多推荐了,返回首页