[关闭]
@HUST-SuWB 2015-02-11T04:36:23.000000Z 字数 2858 阅读 348

重建二叉树

读书笔记


出自《剑指Offer》的面试题6,原题题目为:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输出的前序遍历和中序遍历的结果中都不含重复的数字。
显然,要完成这道题,必须搞清楚一个二叉树的前序遍历,中序遍历之间有什么关系,我们可以通过一个实际的例子来研究。

            3

       /          \

     8             6 

  /      \        /     \

 2       4       7       5

 \                      /  \

     9                 1    0

(由于作业部落暂时不支持上传图片,用外链又一时不知道该先上传到哪去,只能随意画一画了)那么此二叉树对于的前序遍历为:[3-8-2-9-4-6-7-5-1-0],中序遍历为[2-9-8-4-3-76--1-5-0]。我们可以很明显的看到前序遍历的第一个值必然是二叉树的根节点值,那么就可以确定中序遍历中被根节点分隔的正好就是左子树和右子树,然后对于各个左子树和右子树,又可以通过前序遍历的第一个值在中序遍历序列中划分左右子树,这就像一个递归一样。
但是有了这个思路不代表代码能写出来,具体在对数中的节点进行操作的时候,还是很生疏的。我再尝试了一段时间以后,还是没能完整得写出这个题的程序,还是把网上的例子贴过来参考一下吧。
例1:

  1. class Node {
  2. Node left = null;
  3. Node right = null;
  4. char value;
  5. }
  6. public class BinaryTreeBuilder {
  7. /**
  8. * 根据前序遍历和中序遍历重建二叉树子树
  9. * @param preOrder 前序遍历数组
  10. * @param start 子树起始位置
  11. * @param inOrder 中序遍历数组
  12. * @param end 中序遍历结束位置
  13. * @param length 子树节点树
  14. * @return 子树的根节点
  15. */
  16. public static Node buildTree(char[] preOrder, int start,
  17. char[] inOrder, int end, int length) {
  18. //参数验证
  19. if (preOrder == null || preOrder.length == 0 || inOrder == null
  20. || inOrder.length == 0 || length <= 0) {
  21. return null;
  22. }
  23. //建立子树根节点
  24. char value = preOrder[start];
  25. Node root = new Node();
  26. root.value = value;
  27. //递归终止条件:子树只有一个节点
  28. if (length == 1)
  29. return root;
  30. //分拆子树的左子树和右子树
  31. int i = 0;
  32. while (i < length) {
  33. if (value == inOrder[end - i]) {
  34. break;
  35. }
  36. i++;
  37. }
  38. //建立子树的左子树
  39. root.left = buildTree(preOrder, start + 1, inOrder, end - i - 1, length - 1 - i);
  40. //建立子树的右子树
  41. root.right = buildTree(preOrder, start + length - i, inOrder, end, i );
  42. return root;
  43. }
  44. public static void main(String[] args) {
  45. char[] preOrder = new char[] {'a', 'b', 'd', 'c', 'e', 'f'};
  46. char[] inOrder = new char[] {'d', 'b', 'a', 'e', 'c', 'f'};
  47. Node root = buildTree(preOrder, 0, inOrder, inOrder.length - 1, inOrder.length);
  48. }
  49. }

例2:

  1. public class BuildTreePreOrderInOrder {
  2. /**
  3. * Build Binary Tree from PreOrder and InOrder
  4. * _______7______
  5. / \
  6. __10__ ___2
  7. / \ /
  8. 4 3 _8
  9. \ /
  10. 1 11
  11. */
  12. public static void main(String[] args) {
  13. BuildTreePreOrderInOrder build=new BuildTreePreOrderInOrder();
  14. int[] preOrder = {7,10,4,3,1,2,8,11};
  15. int[] inOrder = {4,10,3,1,7,11,8,2};
  16. Node root=build.buildTreePreOrderInOrder(preOrder,0,preOrder.length-1,inOrder,0,preOrder.length-1);
  17. build.preOrder(root);
  18. System.out.println();
  19. build.inOrder(root);
  20. }
  21. public Node buildTreePreOrderInOrder(int[] preOrder,int begin1,int end1,int[] inOrder,int begin2,int end2){
  22. if(begin1>end1||begin2>end2){
  23. return null;
  24. }
  25. int rootData=preOrder[begin1];
  26. Node head=new Node(rootData);
  27. int divider=findIndexInArray(inOrder,rootData,begin2,end2);
  28. int offSet=divider-begin2-1;
  29. Node left=buildTreePreOrderInOrder(preOrder,begin1+1,begin1+1+offSet,inOrder,begin2,begin2+offSet);
  30. Node right=buildTreePreOrderInOrder(preOrder,begin1+offSet+2,end1,inOrder,divider+1,end2);
  31. head.left=left;
  32. head.right=right;
  33. return head;
  34. }
  35. public int findIndexInArray(int[] a,int x,int begin,int end){
  36. for(int i=begin;i<=end;i++){
  37. if(a[i]==x)return i;
  38. }
  39. return -1;
  40. }
  41. public void preOrder(Node n){
  42. if(n!=null){
  43. System.out.print(n.val+",");
  44. preOrder(n.left);
  45. preOrder(n.right);
  46. }
  47. }
  48. public void inOrder(Node n){
  49. if(n!=null){
  50. inOrder(n.left);
  51. System.out.print(n.val+",");
  52. inOrder(n.right);
  53. }
  54. }
  55. }
  56. class Node{
  57. Node left;
  58. Node right;
  59. int val;
  60. public Node(int val){
  61. this.val=val;
  62. }
  63. public Node getLeft(){
  64. return left;
  65. }
  66. public Node getRight(){
  67. return right;
  68. }
  69. public int getVal(){
  70. return val;
  71. }
  72. }
添加新批注
在作者公开此批注前,只有你和作者可见。
回复批注