牛客网的题,重建二叉树。(用树的先序,中序,还原二叉树)

原创 2016年05月20日 21:49:37

题目地址:http://www.nowcoder.com/ta/coding-interviews?page=

题目描述

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

#include<iostream>
#include <vector>
using namespace std;

struct TreeNode {
   int val;
   TreeNode *left;
   TreeNode *right;
   TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

TreeNode* PreInCreat(vector<int> pre,vector<int> in,int l1,int h1,int l2,int h2)
{
    int i;
    TreeNode* root = new TreeNode(pre[l1]);
    for(i=l2;in[i]!=root->val;i++);
    int llen = i-l2;
    int rlen = h2-i;
    if(llen)
        root ->left = PreInCreat(pre,in,l1+1,l1+llen,l2,l2+llen-1);
    else
        root ->left = NULL;
    if(rlen)
        root ->right = PreInCreat(pre,in,h1-rlen+1,h1,h2-rlen+1,h2);
    else
        root ->right = NULL;
    return root;
}

TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> in)
{
    //cout << pre.size();
    return PreInCreat(pre,in,0,pre.size()-1,0,in.size()-1);
}

void PostOrderTravel(TreeNode* L)
{
    if(L->left)
        PostOrderTravel(L->left);
    if(L->right)
        PostOrderTravel(L->right);
    cout << L->val << " ";
}

int main()
{
    vector<int> pre{1,2,3,4,5,6};
    vector<int> in{3,2,1,5,4,6};

    //令i依次为ivec中的每一个元素,并显示
    for(auto i : pre)
        cout << i << " ";
    cout << endl;

    for(auto i : in)
        cout << i << " ";
    cout << endl;

    //reConstructBinaryTree(pre, in);
    TreeNode* root = reConstructBinaryTree(pre, in);
    PostOrderTravel(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:
    struct TreeNode* PreInCreat(vector<int> pre,vector<int> in,int l1,int h1,int l2,int h2)
    {
        int i;
        TreeNode* root = new TreeNode(pre[l1]);
        for(i=l2;in[i]!=root->val;i++);
        int llen = i-l2;
        int rlen = h2-i;
        if(llen)
            root ->left = PreInCreat(pre,in,l1+1,l1+llen,l2,l2+llen-1);
        else
            root ->left = NULL;
        if(rlen)
            root ->right = PreInCreat(pre,in,h1-rlen+1,h1,h2-rlen+1,h2);
        else
            root ->right = NULL;
        return root;
    }
     
    struct TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> in) {
        return PreInCreat(pre,in,0,pre.size()-1,0,in.size()-1);
    }
};

其他人的麻袋

/**
 * 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> in) {
    int length = in.size();
    if (length == 0)
    {
        return NULL;
    }
    vector<int> left_pre, right_pre, left_in, right_in;
    TreeNode *head = new TreeNode(pre[0]); //创建根节点
    int root = 0; //根节点在中序遍历中的位置
    for (int i = 0; i < length; i++)
    {
        if (in[i] == pre[0])
        {
            root = i;
            break;
        }
    }
 
    //对于中序遍历,根节点左边的节点属于左子树,根节点右边的节点属于右子树
    for (int i = 0; i < root; i++)
    {
        left_in.push_back(in[i]);
        left_pre.push_back(pre[i + 1]);
    }
 
    for (int i = root + 1; i < length; i++)
    {
        right_in.push_back(in[i]);
        right_pre.push_back(pre[i]);
    }
 
    head->left = reConstructBinaryTree(left_pre, left_in);
    head->right = reConstructBinaryTree(right_pre, right_in);
 
    return head;
    }
};

继续改改,用左开右闭,[begin,end) 来表示集合

TreeNode* PreInCreat(vector<int> pre,vector<int> in,int l1,int h1,int l2,int h2)
{
    int root_in_in;
    TreeNode* root = new TreeNode(pre[l1]);
    for(root_in_in = l2;in[root_in_in]!=root->val;root_in_in++);

    int llen = root_in_in-l2;
    int rlen = h2 - (root_in_in+1);
    if(llen)
        root->left = PreInCreat(pre,in,l1+1,l1+llen+1,l2,root_in_in);
    else
        root->left = NULL;
    if(rlen)
        //root->right = PreInCreat(pre,in,l1+llen+1,h1,root_in_in+1,h2);
        root->right = PreInCreat(pre,in,h1-rlen,h1,root_in_in+1,h2);
    else
        root->right = NULL;

    return root;
}

TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> in)
{
    //cout << pre.size();
    return PreInCreat(pre,in,0,pre.size(),0,in.size());
}

可以运行


根据先序和中序还原(重建)一个二叉树

package tree.BinaryTree;import java.util.Arrays;/** * 根据先序和中序还原一个二叉树 * * @author yuzhiyun * */...
  • yuzhiyun3536
  • yuzhiyun3536
  • 2016年12月05日 16:06
  • 484

先序中序还原二叉树 和 中序后序还原二叉树 总结

遍历二叉树有三种方法 ①先序:先遍历根结点,后遍历左子树,最后遍历右子树 ②中序:先遍历左子树,后遍历根结点,最后遍历右子树 ③后序:先遍历左子树,后遍历右子树,最后遍历根结点 先序中序还原二...
  • qq_40709140
  • qq_40709140
  • 2017年12月14日 21:42
  • 67

给出先序和中序序列,还原二叉树的规律方法

给出先序和中序序列,还原二叉树的规律方法
  • lws123253
  • lws123253
  • 2016年07月31日 16:03
  • 379

【牛客网】重建二叉树

题目描述: 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,...
  • tiewadhd
  • tiewadhd
  • 2016年08月18日 15:03
  • 271

牛客网刷题--剑指offer(重建二叉树)

题目描述是这样的: 给你一个树的前序遍历和中序遍历,希望你能重建这个二叉树的结构,题目还给了一个例子:某二叉树的前序遍历为{1, 2, 4, 7, 3, 5, 6, 8}和中序遍历{4, 7, 2...
  • liu_sheng_1991
  • liu_sheng_1991
  • 2016年06月29日 16:42
  • 989

牛客网刷题之重建二叉树

题目描述:解题思路:  其实懂得先序排列以及中序排列的规律,用递归还是蛮简单的,不过我还是想了蛮久还看了大神们的代码才ac的,主要是自己数据结构这一块比较薄弱吧!(不薄弱也不会去刷题呀!!!)   ...
  • lks1139230294
  • lks1139230294
  • 2016年10月02日 11:21
  • 275

浙大模拟(1)问题 C: 还原二叉树

题目描述 给一棵二叉树的层序遍历序列和中序遍历序列,求这棵二叉树的先序遍历序列和后序遍历序列。 输入 每个输入文件中一组数据。 第一行一个正整数N(1 输出 输出两行,...
  • linCurry
  • linCurry
  • 2017年03月12日 12:04
  • 203

二叉树三种遍历及还原

图一 先序:根左子树 右子树 从头结点开始ABDECF 规律:最左边的元素是这个二叉树最顶层节点。也就是A 中序:左子树根 右子树 从最后一层叶子节点开始遍历:DBEAFC 规律:从整体来讲A在树...
  • hemerry
  • hemerry
  • 2016年10月28日 17:54
  • 992

给出先序序列,中序序列恢复二叉树

//给出先序序列,中序序列恢复出二叉树的方法 //方法1:利用递归的方法,首先先找出根节点,然后有先序和后序的特点可知:先序序列中在根节点左边的是左子树,根节点右面的是右子树 //有此来,进行递归...
  • qq_28633157
  • qq_28633157
  • 2015年11月15日 21:14
  • 444

牛客网的题,重建二叉树。(用树的先序,中序,还原二叉树)

题目描述 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7...
  • wsyxhwj
  • wsyxhwj
  • 2016年05月20日 21:49
  • 856
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:牛客网的题,重建二叉树。(用树的先序,中序,还原二叉树)
举报原因:
原因补充:

(最多只允许输入30个字)