第3章 结构之法——重建二叉树

 1 package chapter3jiegouzhifa.RebuildBinTree;
 2 
 3 /**
 4  * 重建二叉树
 5  * 递归解法
 6  * @author DELL
 7  *
 8  */
 9 public class RebuildBinTree {
10     //定义节点类
11     public static class Node{
12         int left;  //左子树位置
13         int right;  //右子树位置
14         char value;  //节点值
15         public Node(int left, int right, char value){
16             this.left = left;
17             this.right = right;
18             this.value = value;
19         }
20         public void setLeft(int left) {
21             this.left = left;
22         }
23         public void setRight(int right) {
24             this.right = right;
25         }
26     }
27     /**
28      * 重建二叉树
29      * @param preOrder 前序遍历结果
30      * @param inOrder    中序遍历结果
31      * @param tree    重建的树
32      * @param root  当前重建树的根节点位置
33      */
34     public static void rebuild(String preOrder, String inOrder, Node[] tree, int root){
35         
36         int nTreeLen = preOrder.length();
37         //检查边界条件
38         if(preOrder.length()==0||inOrder.length()==0)
39             return ;
40         //获取当前遍历的第一个节点
41         Node temp = new Node(-1,-1,preOrder.charAt(0));
42         //如果节点为空，把当前节点复制到根节点
43         if(tree[root]==null){
44             tree[root] = temp;
45         }
46         //如果当前树的长度为1，那么已经是最后一个节点
47         if(nTreeLen == 1){    
48             return ;
49         }
50         //寻找左子树的结尾
51         int i=0;
52         while(inOrder.charAt(i)!=temp.value&&i<nTreeLen){
53             i++;
54         }
55 //        System.out.println(i);
56         //重建左子树
57         int index = root;
58         if(i>0){
59             tree[index].setLeft(++root);
60             Node left = new Node(-1,-1,preOrder.charAt(1));
61             tree[root]=left;
62 //            System.out.println(preOrder.substring(1,i+1)+" "+i);
63             rebuild(preOrder.substring(1,i+1),inOrder.substring(0, i),tree,root);
64         }
65         //重建右子树
66         if(nTreeLen-i-1>0){
67             tree[index].setRight(root+i);
68             Node right = new Node(-1,-1,preOrder.charAt(i+1));
69             tree[root+i] = right;
70             rebuild(preOrder.substring(i+1,nTreeLen),inOrder.substring(i+1, nTreeLen),tree,root+i);
71         }
72     }
73     public static void main(String[] args) {
74         String preOrder = "abdcef";
75         String inOrder = "dbaecf";
76         Node[] tree = new  Node[preOrder.length()];
77         rebuild(preOrder,inOrder,tree,0);
78         System.out.println("重建的树为：");
79         for(int i=0;i<tree.length;i++){
80             String left,right;
81             if(tree[i].left!=-1)
82                 left = String.valueOf(tree[tree[i].left].value);
83             else
84                 left = null;
85             if(tree[i].right!=-1)
86                 right = String.valueOf(tree[tree[i].right].value);
87             else
88                 right = null;
89             System.out.println(tree[i].value+" 左子树："+left+" 右子树："+right);
90         }
91     }
92     
93 }

重建的树为：
a 左子树：b 右子树：c
b 左子树：d 右子树：null
d 左子树：null 右子树：null
c 左子树：e 右子树：f
e 左子树：null 右子树：null
f 左子树：null 右子树：null