算法-重建二叉树

问题描述:

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

我自己的答案:

package com.zjq.arithmetic;

import java.util.ArrayList;
import java.util.List;

/**
* 题目描述
* 输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。
* 假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
* 例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，
* 则重建二叉树并返回。
* @author zjq
* @Date 2017/08/08
*
*/

public class BinaryTree {

public static void main(String[] args) {
int[] a = {1,2,4,7,3,5,6,8};
int[] b = {4,7,2,1,5,3,8,6};
BinaryTree binaryTree=new BinaryTree();
binaryTree.reConstructBinaryTree(a, b);

}

public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
TreeNode treeNode;
if(pre.length>=1) {
int midNode=pre[0];
List<Integer> inL=new ArrayList<Integer>();
List<Integer> inR=new ArrayList<Integer>();
List<Integer> preL=new ArrayList<Integer>();
List<Integer> preR=new ArrayList<Integer>();
int flag=0;
treeNode=new TreeNode(midNode);
for(int i=0;i<=in.length-1;i++) {
if(in[i]==midNode) {
flag=1;
}else {
if(flag==0) {
inL.add(in[i]);
}else {
inR.add(in[i]);
}
}
}
for(int i=1;i<=inL.size();i++) {
preL.add(pre[i]);
}

for(int i=inL.size()+1;i<pre.length;i++) {
preR.add(pre[i]);
}
if(preL.size()>0) {
int[] newPre=new int[preL.size()];
int[] newIn=new int[inL.size()];
for(int i=0;i<preL.size();i++) {
newPre[i]=preL.get(i);
}
for(int i=0;i<inL.size();i++) {
newIn[i]=inL.get(i);
}

treeNode.left=reConstructBinaryTree(newPre,newIn);
}
if(preR.size()>0) {
int[] newPre=new int[preR.size()];
int[] newIn=new int[inR.size()];
for(int i=0;i<preR.size();i++) {
newPre[i]=preR.get(i);
}
for(int i=0;i<inR.size();i++) {
newIn[i]=inR.get(i);
}

treeNode.right=reConstructBinaryTree(newPre,newIn);
}
return treeNode;
}
return null;

}

}

class TreeNode {
int val;
TreeNode left;
TreeNode right;

TreeNode(int x) {
val = x;
}
}


更好的解答方法:
链接：https://www.nowcoder.com/questionTerminal/8a19cbe657394eeaac2f6ea9b0f6fcf6
来源：牛客网

public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        TreeNode root=reConstructBinaryTree(pre,0,pre.length-1,in,0,in.length-1);
        return root;
    }
    //前序遍历{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}
    private TreeNode reConstructBinaryTree(int [] pre,int startPre,int endPre,int [] in,int startIn,int endIn) {
         
        if(startPre>endPre||startIn>endIn)
            return null;
        TreeNode root=new TreeNode(pre[startPre]);
         
        for(int i=startIn;i<=endIn;i++)
            if(in[i]==pre[startPre]){
                root.left=reConstructBinaryTree(pre,startPre+1,startPre+i-startIn,in,startIn,i-1);
                root.right=reConstructBinaryTree(pre,i-startIn+startPre+1,endPre,in,i+1,endIn);
            }
                 
        return root;
    }
}