zoukankan      html  css  js  c++  java
  • 重建二叉树

    题目来源https://www.acwing.com/problem/content/23/

    思路:这是很经典的题,根据前序遍历创建根节点,再通过递归创建左右子树,最后根节点指向左右子树。

    具体做法是通过在中序遍历中找到根节点的位置k,分别得到中序遍历左右子树的区间,得到左右区间的长度l,r。

    而前序遍历的左子树位置是从根节点接着的l个节点,右子树区间是左子树的下一个节点到最后的节点。

    中序遍历遍历找位置用个map即可。由于每个节点只创建一次,时间复杂度o(n)。

    /**
     * Definition for a binary tree node.
     * struct TreeNode {
     *     int val;
     *     TreeNode *left;
     *     TreeNode *right;
     *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
     * };
     */
    class Solution {
    public:
        map<int,int>m;
        vector<int>preorder,inorder;
        TreeNode* buildTree(vector<int>& _preorder, vector<int>& _inorder) {
            preorder=_preorder,inorder=_inorder;
            for(int i=0;i<_inorder.size();i++){
                m[_inorder[i]]=i;
            }        
            return dfs(0,_preorder.size()-1,0,_inorder.size()-1);
        }
        TreeNode* dfs(int pl,int pr,int il,int ir){
            if(pl>pr)return NULL;
            TreeNode* root=new TreeNode(preorder[pl]);
            int k=m[root->val];
            auto left=dfs(pl+1,pl+1+k-il-1,il,k-1);
            auto right=dfs(pl+k-il+1,pr,k+1,ir);
            root->left=left;root->right=right;
            return root;
        }
    };

     

  • 相关阅读:
    一些$LCT$的瓜皮题目
    写点东西(关于背包问题)
    字符串算法总结
    常系数齐次线性递推
    原根算法与剩余定理
    问题集
    常用链接
    回形针PaperClip
    6.824拾遗
    杂项
  • 原文地址:https://www.cnblogs.com/mohari/p/13739118.html
Copyright © 2011-2022 走看看