程序员基本功系列5——二叉树

1、二叉树基础

1.1、树的几个概念

    节点的高度:节点到叶子节点的最大路径(边数)

    节点的深度:根节点到这个节点所经历的边数

    节点的层数:节点的深度+1

    树的高度:根节点的高度

  高度和深度的计数都是从0开始,来看个例子:

      

1.2、满二叉树和完全二叉树

(1)满二叉树

    叶子节点全都在最后一层,除叶子节点外,每个节点都有左右两个节点,这种二叉树就是满二叉树。

      

     其中2号就是满二叉树。

(2)完全二叉树

    叶子节点都在最底下两层,最后一层的叶子节点都靠左排列,并且除了叶子节点,每个节点都有左右两个节点,这种二叉树就是完全二叉树。上图中3号就是一棵完全二叉树。

    再来看几个完全二叉树和非完全二叉树的例子:

      

    和满二叉树一样,除了叶子节点其他节点都达到最大,这一点很好理解,那为什么完全二叉树的要求最后一层都靠左排列呢?

    要明白这一点就要看二叉树的存储方式?主要有两种:链式存储和基于数组的顺序存储。

  链式存储:这是最长用的二叉树表示方式,每个节点三个字段,一个存储数据,另外两个是指向左右节点的指针。

      

  基于数组的顺序存储:把根节点存储在下标 i = 1 的位置,那左子节点存储在下标 2 * i = 2 的位置,右子节点存储在 2 * i + 1 = 3 的位置。以此类推,B 节点的左子节点存储在 2 * i = 2 * 2 = 4 的位置,右子节点存储在 2 * i + 1 = 2 * 2 + 1 = 5 的位置。

       

    完全二叉树叶子节点都靠左排列就是为了节省存储空间。

1.3、二叉树的遍历

  经典的三种方式:前序遍历、中序遍历、后序遍历。

    前序遍历是指,对于树中的任意节点来说,先打印这个节点,然后再打印它的左子树,最后打印它的右子树。

    中序遍历是指,对于树中的任意节点来说,先打印它的左子树,然后再打印它本身,最后打印它的右子树。

    后序遍历是指,对于树中的任意节点来说,先打印它的左子树,然后再打印它的右子树,最后打印这个节点本身 

  实际上,前中后序遍历就是递归过程,我们直接看代码:

void preOrder(Node* root) {
  if (root == null) return;
  print root // 此处为伪代码,表示打印root节点
  preOrder(root->left);
  preOrder(root->right);
}

void inOrder(Node* root) {
  if (root == null) return;
  inOrder(root->left);
  print root // 此处为伪代码,表示打印root节点
  inOrder(root->right);
}

void postOrder(Node* root) {
  if (root == null) return;
  postOrder(root->left);
  postOrder(root->right);
  print root // 此处为伪代码,表示打印root节点
}

  三种遍历,每个节点最多会被访问两次,与节点个数正相关,所以时间复杂度是 O(n)。

2、二叉查找树

  二叉查找树是二叉树中最常用的一种类型,也就二叉搜索树。它支持快速查找、插入和删除数据。

  二叉查找树要求,在树中的任意一个节点,其左子树中的每个节点的值,都要小于这个节点的值,而右子树节点的值都大于这个节点的值。

2.1、二叉查找树的操作

(1)二叉查找树的查找操作

  先取根节点,如果它等于我们要查找的数据,那就返回。如果要查找的数据比根节点的值小,那就在左子树中递归查找;如果要查找的数据比根节点的值大,那就在右子树中递归查找。

public class BinarySearchTree {
  private Node tree;
public static class Node { private int data; private Node left; private Node right; public Node(int data) { this.data = data; } } public Node find(int data) { Node p = tree; while (p != null) { if (data < p.data) p = p.left; else if (data > p.data) p = p.right; else return p; } return null; } }

(2)二叉查找树的插入操作

  二叉查找树的插入过程有点类似查找操作。新插入的数据一般都是在叶子节点上,所以我们只需要从根节点开始,依次比较要插入的数据和节点的大小关系。如果要插入的数据比节点的数据大,并且节点的右子树为空,就将新数据直接插到右子节点的位置;如果不为空,就再递归遍历右子树,查找插入位置。同理,如果要插入的数据比节点数值小,并且节点的左子树为空,就将新数据插入到左子节点的位置;如果不为空,就再递归遍历左子树,查找插入位置。

public void insert(int data) {
  if (tree == null) {
    tree = new Node(data);
    return;
  }

  Node p = tree;
  while (p != null) {
    if (data > p.data) {
      if (p.right == null) {
        p.right = new Node(data);
        return;
      }
      p = p.right;
    } else { // data < p.data
      if (p.left == null) {
        p.left = new Node(data);
        return;
      }
      p = p.left;
    }
  }
}

(3)二叉查找树的删除操作

  删除操作比较复杂,因为可能涉及到重构,根据要删除节点的子节点个数不同,分三种情况考虑:

    

  • 如果要删除的节点没有子节点,只需要直接将父节点中,指向要删除节点的指针置为 null。比如图中的删除节点 55。

  • 如果要删除的节点只有一个子节点(只有左子节点或者右子节点),只需要更新父节点中指向要删除节点的指针,让它指向要删除节点的子节点就可以了。比如图中的删除节点 13。

  • 如果要删除的节点有两个子节点,这就比较复杂了。需要找到这个节点的右子树中的最小节点,把它替换到要删除的节点上。然后再删除掉这个最小节点,因为最小节点肯定没有左子节点(如果有左子结点,那就不是最小节点了),所以,可以应用上面两条规则来删除这个最小节点。比如图中的删除节点 18。

public void delete(int data) {
  Node p = tree; // p指向要删除的节点,初始化指向根节点
  Node pp = null; // pp记录的是p的父节点
  while (p != null && p.data != data) {
    pp = p;
    if (data > p.data) p = p.right;
    else p = p.left;
  }
  if (p == null) return; // 没有找到

  // 要删除的节点有两个子节点
  if (p.left != null && p.right != null) { // 查找右子树中最小节点
    Node minP = p.right;
    Node minPP = p; // minPP表示minP的父节点
    while (minP.left != null) {
      minPP = minP;
      minP = minP.left;
    }
    p.data = minP.data; // 将minP的数据替换到p中
    p = minP; // 下面就变成了删除minP了
    pp = minPP;
  }

  // 删除节点是叶子节点或者仅有一个子节点
  Node child; // p的子节点
  if (p.left != null) child = p.left;
  else if (p.right != null) child = p.right;
  else child = null;

  if (pp == null) tree = child; // 删除的是根节点
  else if (pp.left == p) pp.left = child;
  else pp.right = child;
}

(4)二叉查找树的其他操作

  除了插入、删除、查找操作之外,二叉查找树中还可以支持快速地查找最大节点和最小节点、前驱节点和后继节点。

  另外还有一个重要特性,二叉查找树的中序遍历可以输出有序的数据序列,时间复杂度 O(n),非常高效。

(5)时间复杂度分析

  二叉查找树的查找、插入和删除的时间复杂度跟跟树的高度成正比,所以时间复杂度就是 O(height)。在极端情况下,二叉查找树会退化成链表,时间复杂度就是 O(n),如果二叉树是平衡二叉树(3节讲),平衡二叉树的树高接近 O(logn),

所以对于平衡二叉树来说时间复杂度就是 O(logn)。

2.2、支持重复数据的二叉查找树

   很多时候,在实际的开发中,在二叉查找树中存储的,是一个包含很多字段的对象。利用对象的某个字段作为键值(key)来构建二叉查找树。把对象中的其他字段叫作卫星数据。

  上面小节中所提到的操作都是针对不存在重复键值的情况,那如果存在重复键值怎么解决呢?几种思路:

(1)二叉树的每个节点不仅存储一个数据,可以通过链表和支持动态扩容的数组等数据结构,把相同键值的数据存储在一个节点上。

(2)每个节点还是只存储一个数据,如果遇到相同的键值,将要插入的数据放到右子树,也就是把相同的键值按大于这个节点的值来处理。

    当要查找数据的时候,遇到值相同的节点,并不停止查找操作,而是继续在右子树中查找,直到遇到叶子节点,才停止。这样就可以把键值等于要查找值的所有节点都找出来。

      

(3)对于要删除的数据,并不真正删除,只是将节点标记为删除状态,这样就避免删除带来的复杂性,但是比较浪费存储空间,并且当数据量越大,性能会越低。

3、二叉树的深度优先和广度优先遍历

(1)深度优先遍历  

  深度优先遍历(DFS):从根节点出发,沿着左子树方向进行纵向遍历,直到找到叶子节点为止。然后回溯到前一个节点,进行右子树节点的遍历,直到遍历完所有可达节点为止。 

  深度优先遍历可以借助栈(先进后出)的特性来实现。

    public void depthOrderTraversal(){
        if(root==null){
            return;
        }       
        Stack<TreeNode> stack = new Stack();   //用栈实现
        stack.push(root);       
        while(stack.isEmpty()==false){
            TreeNode node=stack.pop();
            if(node.right!=null){
                stack.push(node.right);
            }
            if(node.left!=null){
                stack.push(node.left);
            }           
        }
    }

  leetcode:二叉树中和为某一值得路径

(2)广度优先遍历

  广度优先遍历:从根节点出发,在横向遍历二叉树层段节点的基础上纵向遍历二叉树的层次。

  广度优先遍历可以借助队列(先进先出)的特性来实现。 

    public void levelOrderTraversal(){
        if(root==null){
            return;
        }
        ArrayDeque<TreeNode> queue=new ArrayDeque<TreeNode>();
        queue.add(root);
        while(queue.isEmpty()==false){
            TreeNode node=queue.remove();
            if(node.left!=null){
                queue.add(node.left);
            }
            if(node.right!=null){
                queue.add(node.right);
            }
        }
    }

4、平衡二叉树之红黑树

  平衡二叉树严格的定义:除了二叉查找树的特点,二叉树中任意节点左右子树的高度相差不能大于1。上面提到的完全二叉树和满二叉树就是严格意义上的平衡二叉树。

  AVL树也是严格意义上的平衡二叉树,但是很多平衡二叉树并没有严格符合上面的定义(即任意节点左右子树的高度相差不能大于1),比如下面要说的红黑树。这里我们就要理解平衡二叉树中“平衡”的意义:其实就是让整棵树左右看起来比较“对称”、比较“平衡”,不要出现左子树很高、右子树很矮的情况。这样就能让整棵树的高度相对来说低一些,相应的插入、删除、查找等操作的效率高一些。

  所以只要树高不比logn大很多,也可以认为符合平衡二叉树。

  那我们来看红黑树,顾名思义,它的一类节点被标记为红色,一类节点被标记为黑色,此外还要满足这样几个要求:

    • 根节点是黑色的

    • 每个叶子节点都是黑色的空节点(NULL),也就是说叶子节点不存储数据。这一点主要是为了简化红黑树代码实现而设置

    • 任何相邻的节点都不能同时为红色,也就是说红色节点是被黑色节点隔开的

    • 每个节点,从该节点到达其可达叶子节点的所有路径,都包含相同数目的黑色节点

  来看一下红黑树的例子,这里去掉叶子节点:

      

  这里思考一个问题:AVL树是严格的平衡二叉树,查询性能非常高,为什么红黑树在实际开发中使用更多呢?

    其实有利有弊,AVL树为了维持严格平衡,在插入和删除时都要做调整,这样就增加了复杂度和性能开销,但是红黑树的近似平衡,就使得插入、删除和查询等各种操作的性能比较稳定。

  红黑树的实现较为复杂,在实际开发或是面试中也不太会自己写一个红黑树,有的放矢地学习,这里红黑树的实现就先略过了。 

posted @ 2022-02-14 20:23  jingyi_up  阅读(99)  评论(0编辑  收藏  举报