题目描述:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:前序序列可以看做是重建二叉树时插入节点的顺序,中序序列中每一个结点的左边的结点,都是这个结点的左子树,右边的结点是这个结点的右子树,因此可以根据中序遍历判断该结点有没有左结点和右结点。编写一个递归函数实现给二叉树插入一个结点,并判断它有无左右结点,进行递归。
ac代码:
/** * Definition for binary tree * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ import java.util.ArrayList; import java.util.Collections; import java.util.List; import java.util.Scanner; public class Solution { ArrayList<Integer> list1=new ArrayList<Integer>(); ArrayList<Integer> list2=new ArrayList<Integer>(); public TreeNode reConstructBinaryTree(int [] pre,int [] in) { for(int i=0;i<pre.length;i++) { list1.add(pre[i]); list2.add(in[i]); } TreeNode node=null; node=dfs(0,pre.length-1); return node; } public TreeNode dfs(int l,int r) { if(list1.isEmpty()) return null; int x=list1.get(0); list1.remove(0); TreeNode node=new TreeNode(x); int k=0; for(int i=l;i<=r;i++) { if(list2.get(i)==x) k=i; } if(k==l) node.left=null; else node.left=dfs(l,k-1); if(k==r) node.right=null; else node.right=dfs(k+1,r); return node; } }