二叉树总结（创建二叉树）

二叉树总结（创建二叉树）

• 根据前序遍历和中序遍历创建二叉树（注意根据前、中、后序列创建唯一二叉树的时候，一定要有两个遍历序列，且其中一个必须为中序。

• 我们先定义二叉树的节点结构，一般包含左孩子，右孩子，键值等参数信息。

  /**
   * 二叉树节点结构
   */
  class TreeNode{
      int val;       //节点值
      TreeNode left; //左孩子
      TreeNode right;//右孩子
  
      public TreeNode(int val) {
          this.val = val;
      }
  
      public TreeNode(int val, TreeNode left, TreeNode right) {
          this.val = val;
          this.left = left;
          this.right = right;
      }
  }
  

• 根据前序和中序序列遍历二叉树（中序和后序序列生成二叉树同理）

  /**
   * 根据先序和中序遍历序列生成二叉树
   * 只要先序，中序，后序中的两个序列
   * 就能生成一个二叉树，这里以先序和
   * 中序为例
   * @param pre       先序序列数组
   * @param preStart  先序序列的开始下标
   * @param preEnd    先序序列的结束下标
   * @param mid       中序序列数组
   * @param midStart  中序序列的开始下标
   * @param midEnd    中序序列的结束下标
   * @return          生成树的头结点
   */
  public TreeNode creatBinaryTreeByPreMid(int[] pre,int preStart,int preEnd,int mid[],int midStart,int midEnd){
      //判断结束标记
      if (preStart>preEnd||midStart>midEnd){
          return null;
      }
      //根据先序遍历得到根节点（因为下面使用的是递归，所以这个根节点不单单指的是整棵树的根节点）
      TreeNode rootNode = new TreeNode(pre[preStart]);
      //获得根节点在中序遍历的下标位置
      int location = 0;
      for (int i = midStart; i <= midEnd ; i++) {
          if (mid[i]==rootNode.val){
              location = i;
          }
      }
      //两者的偏移量（计算左子树有几个几点几点树减一就是偏移量）
      int offSet = location - midStart - 1;
      //递归获得当前节点的左孩子
      TreeNode left = creatBinaryTreeByPreMid(pre, preStart + 1, preStart+offSet+1, mid, midStart,midStart + offSet);
      TreeNode right = creatBinaryTreeByPreMid(pre, preStart + offSet + 2, preEnd, mid, location + 1, midEnd);
      rootNode.left = left;
      rootNode.right = right;
      return rootNode;
  }
  

• 除了根据两个遍历序列生成二叉树之外，还可以根据层次遍历生成唯一二叉树，但注意层次遍历的二叉树一定是完全二叉树。

  /**
   * 根据分层遍历生成二叉树
   * @param array     分层遍历的序列数组
   * @return          生成树的头结点
   */
  public TreeNode createBinaryTreeByLayerErgodic(int[] array){
      ArrayList<TreeNode> treeNodes = new ArrayList<>(); //用于存放树的节点
      //根据数组生成树的初始节点
      for (int i = 0; i < array.length; i++) {
          TreeNode treeNode = new TreeNode(array[i], null, null);
          treeNodes.add(treeNode);//生成的节点存放到ArrayList中
      }
  
      //判断传过来的数组是否为空数组
      if (treeNodes.size()>0){
          //这里只判断到倒数第二个非叶子节点
          for (int i = 0; i < (array.length/2-1); i++) {
              //判断左节点是否为空
              if (treeNodes.get(2*i+1)!=null){
                  treeNodes.get(i).left = treeNodes.get(2*i+1);
              }
              //判断右节点是否为空
              if (treeNodes.get(2*i+2)!=null){
                  treeNodes.get(i).right = treeNodes.get(2*i+2);
              }
          }
      }else {
          return null;
      }
      //处理最后一个非叶子节点
      int lastRootNode = treeNodes.size() / 2 - 1;//获得最后一个非叶子节点
      //左孩子
      treeNodes.get(lastRootNode).left = treeNodes.get(2*lastRootNode+1);
      //判断是否存在右孩子
      if (treeNodes.size()%2==1){
          treeNodes.get(lastRootNode).right = treeNodes.get(2*lastRootNode+2);
      }
      return treeNodes.get(0);
  }