zoukankan      html  css  js  c++  java
  • 【剑指offer04重建二叉树】

    题目描述

    输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
     
    通过前序的第一个点,对中序切分->左半边和右半边,然后对前序也切分为左半边和右半边,递归
    /**
     * Definition for binary tree
     * struct TreeNode {
     *     int val;
     *     TreeNode *left;
     *     TreeNode *right;
     *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
     * };
     */
    class Solution {
    
    public:
    
        struct TreeNode *reConstructBinaryTree(vector<int> pre, vector<int> vin) {
            int vinlen = vin.size();
            if (pre.size() == 0) //pre 和 vin 是等长的  用哪个都可以
                return NULL;
    
            vector<int> left_pre, right_pre, left_in, right_in;
    
            //创建根节点,根节点肯定是前序遍历的第一个数
            TreeNode *head = new TreeNode(pre[0]);
    
            //找到中序遍历根节点所在位置,存放于变量gen中
            int gen_now = 0;
            for (int i = 0; i < vinlen; i++) {
                if (vin[i] == pre[0]) {
                    gen_now = i;
                    break;
                }
            }
    
            //对于中序遍历,根节点左边的节点位于二叉树的左边,根节点右边的节点位于二叉树的右边
            //利用上述这点,对二叉树节点进行归并
            for (int i = 0; i < gen_now; i++) {
                left_in.push_back(vin[i]);
                left_pre.push_back(pre[i + 1]);//把属于左子树的加进去
            }
            //右边
            for (int i = gen_now + 1; i < vinlen; i++) {
                right_in.push_back(vin[i]);
                right_pre.push_back(pre[i]);
            }
    
            //和shell排序的思想类似,取出前序和中序遍历根节点左边和右边的子树
            //递归,再对其进行上述所有步骤,即再区分子树的左、右子子数,直到叶节点
    
            head->left = reConstructBinaryTree(left_pre, left_in);
            head->right = reConstructBinaryTree(right_pre, right_in);
    
            return head;
    
        }
    
    };
  • 相关阅读:
    Solr4.8.0源码分析(16)之SolrCloud索引深入(3)
    Solr4.8.0源码分析(15) 之 SolrCloud索引深入(2)
    Solr4.8.0源码分析(14)之SolrCloud索引深入(1)
    Solr In Action 笔记(4) 之 SolrCloud分布式索引基础
    Solr In Action 笔记(3) 之 SolrCloud基础
    change buffer太复杂了
    mtr语言真是逆天了
    mysql自动化测试第一个例子
    mysql mtr写入数据
    垃圾规范
  • 原文地址:https://www.cnblogs.com/Stephen-Jixing/p/13124001.html
Copyright © 2011-2022 走看看