zoukankan      html  css  js  c++  java
  • 面试题15:有前或者后续遍历和中序遍历构建二叉树

    二叉树的构建必须要有中序遍历,另外还需要前序和后续中的一种。下面分别是有前序和中序、后续和中序采用递归方法构建二叉树的过程

    /**
     * Definition for binary tree
     * struct TreeNode {
     *     int val;
     *     TreeNode *left;
     *     TreeNode *right;
     *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
     * };
     */
    class Solution {
    public:
        TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) {
            if(preorder.empty()) return nullptr;
            int n = preorder.size();
            int rootVal = preorder[0];
    
            TreeNode* root = new TreeNode(rootVal);
            vector<int> leftIn(inorder.begin(),find(inorder.begin(),inorder.end(),rootVal));
            int leftSize = leftIn.size();
            vector<int> leftPre(preorder.begin()+1,preorder.begin()+1+leftSize);
    
            vector<int> rightIn(inorder.begin()+leftSize+1,inorder.end());
            vector<int> rightPre(preorder.begin()+leftSize+1,preorder.end());
            root->left= buildTree(leftPre,leftIn);
            root->right = buildTree(rightPre,rightIn);
    
            return root;
        }
    };
    /**
     * Definition for binary tree
     * struct TreeNode {
     *     int val;
     *     TreeNode *left;
     *     TreeNode *right;
     *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
     * };
     */
    class Solution{
    public:
        TreeNode* buildTree(vector<int>& inorder,vector<int>& postOrder){
            if(postOrder.empty()) return nullptr;
    
            int n = postOrder.size();
            int rootVal = postOrder[n-1];
            TreeNode* root = new TreeNode(rootVal);
    
            vector<int> leftIn(inorder.begin(),find(inorder.begin(),inorder.end(),rootVal));
            int leftSize = leftIn.size();
            vector<int> leftPost(postOrder.begin(),postOrder.begin()+leftSize);
    
            vector<int> rightIn(inorder.begin()+leftSize+1,inorder.end());
            vector<int> rightPost(postOrder.begin()+leftSize,postOrder.end()-1);
    
            root->left = buildTree(leftIn,leftPost);
            root->right = buildTree(rightIn,rightPost);
            return root;
        }
    };
  • 相关阅读:
    Spring阶段性学习:基础、配置解析、回调
    谈谈Spring对于@Configuration的Cglib代理
    IDEA使用Docker插件部署jar
    实用的IDEA插件
    Alibaba Cloud Toolkit插件部署jar
    谈谈Spring的ImportSelector和ImportBeanDefinitionRegistrar
    谈谈Spring的ConfigurationClassPostProcessor
    Mariadb使用总结
    分布式开放消息系统(RocketMQ)的原理与实践
    redis学习笔记
  • 原文地址:https://www.cnblogs.com/wxquare/p/6858421.html
Copyright © 2011-2022 走看看