当前位置: 首页 > 编程笔记 >

java算法实现红黑树完整代码示例

尉迟跃
2023-03-14
本文向大家介绍java算法实现红黑树完整代码示例,包括了java算法实现红黑树完整代码示例的使用技巧和注意事项,需要的朋友参考一下

红黑树

定义

红黑树(英语:Red–black tree)是一种自平衡二叉查找树,是在计算机科学中用到的一种数据结构,典型的用途是实现关联数组。

红黑树的另一种定义是含有红黑链接并满足下列条件的二叉查找树:

红链接均为左链接;没有任何一个结点同时和两条红链接相连;该树是完美黑色平衡的,即任意空链接到根结点的路径上的黑链接数量相同。

满足这样定义的红黑树和相应的2-3树是一一对应的。

旋转

旋转又分为左旋和右旋。通常左旋操作用于将一个向右倾斜的红色链接旋转为向左链接。对比操作前后,可以看出,该操作实际上是将红线链接的两个节点中的一个较大的节点移动到根节点上。

左旋操作如下图:

右旋旋操作如下图:

即:

复杂度

红黑树的平均高度大约为lgN。

下图是红黑树在各种情况下的时间复杂度,可以看出红黑树是2-3查找树的一种实现,他能保证最坏情况下仍然具有对数的时间复杂度。

Java代码

import java.util.NoSuchElementException;
import java.util.Scanner;
public class RedBlackBST<key extends="" key="">, Value> {
  private static final boolean RED = true;
  private static final boolean BLACK = false;
  private Node root; //root of the BST
  private class Node {
    private Key key;      //key
    private Value val;     //associated data
    private Node left, right;  //links to left and right subtrees
    private boolean color;   //color of parent link
    private int size;      //subtree count
    public Node(Key key, Value val, boolean color, int size) {
      this.key = key;
      this.val = val;
      this.color = color;
      this.size = size;
    }
  }
  //is node x red?
  private boolean isRed(Node x) {
    if(x == null) {
      return false;
    }
    return x.color == RED;
  }
  //number of node in subtree rooted at x; 0 if x is null
  private int size(Node x) {
    if(x == null) {
      return 0;
    }
    return x.size;
  }  
  /**
   * return the number of key-value pairs in this symbol table
   * @return the number of key-value pairs in this symbol table
   */
  public int size() {
    return size(root);
  }
  /**
   * is this symbol table empty?
   * @return true if this symbol table is empty and false otherwise
   */
  public boolean isEmpty() {
    return root == null;
  }
  /**
   * return the value associated with the given key
   * @param key the key
   * @return the value associated with the given key if the key is in the symbol table, and null if it is not.
   */
  public Value get(Key key) {
    if(key == null) {
      throw new NullPointerException("argument to get() is null");
    }
    return get(root, key);
  }
  //value associated with the given key in subtree rooted at x; null if no such key
  private Value get(Node x, Key key) {
    while(x != null) {
      int cmp = key.compareTo(x.key);
      if(cmp < 0) {
        x = x.left;
      }
      else if(cmp > 0) {
        x = x.right;
      }
      else {
        return x.val;
      }        
    }
    return null;
  }
  /**
   * does this symbol table contain the given key?
   * @param key the key
   * @return true if this symbol table contains key and false otherwise
   */
  public boolean contains(Key key) {
    return get(key) != null;
  }
  /***************************************************************************
  * Red-black tree insertion.
  ***************************************************************************/
  /**
   * Inserts the specified key-value pair into the symbol table, overwriting the old 
   * value with the new value if the symbol table already contains the specified key.
   * Deletes the specified key (and its associated value) from this symbol table
   * if the specified value is null.
   *
   * @param key the key
   * @param val the value
   * @throws NullPointerException if key is null
   */
  public void put(Key key, Value val) {
    if (key == null) {
      throw new NullPointerException("first argument to put() is null");
    }
    if (val == null) {
      delete(key);
      return;
    }
    root = put(root, key, val);
    root.color = BLACK;    
  }
  // insert the key-value pair in the subtree rooted at h
  private Node put(Node h, Key key, Value val) {
    if(h == null) {
      return new Node(key, val, RED, 1);
    }
    int cmp = key.compareTo(h.key);
    if(cmp < 0) {
      h.left = put(h.left, key, val);
    }
    else if(cmp > 0) {
      h.right = put(h.right, key, val);
    }
    else {
      h.val = val;
    }
    if(isRed(h.right) && !isRed(h.left)) {
      h = rotateLeft(h);
    }
    if(isRed(h.left) && isRed(h.left.left)) {
      h = rotateRight(h);
    }
    if(isRed(h.left) && isRed(h.right)) {
      flipColors(h);
    }
    h.size = size(h.left) + size(h.right) + 1;
    return h;
  }
  /***************************************************************************
  * Red-black tree deletion.
  ***************************************************************************/
 
  /**
   * Removes the smallest key and associated value from the symbol table.
   * @throws NoSuchElementException if the symbol table is empty
   */
  public void deleteMin() {
    if (isEmpty()) {
      throw new NoSuchElementException("BST underflow");
    }
    // if both children of root are black, set root to red
    if (!isRed(root.left) && !isRed(root.right))
      root.color = RED;
    root = deleteMin(root);
    if (!isEmpty()) root.color = BLACK;
    // assert check();
  }
  // delete the key-value pair with the minimum key rooted at h
  // delete the key-value pair with the minimum key rooted at h
  private Node deleteMin(Node h) { 
    if (h.left == null){
      return null;
    }
    if (!isRed(h.left) && !isRed(h.left.left)) {
      h = moveRedLeft(h);
    }
    h.left = deleteMin(h.left);
    return balance(h);
  }
  /**
   * Removes the largest key and associated value from the symbol table.
   * @throws NoSuchElementException if the symbol table is empty
   */
  public void deleteMax() {
    if (isEmpty()) {
      throw new NoSuchElementException("BST underflow");
    }
    // if both children of root are black, set root to red
    if (!isRed(root.left) && !isRed(root.right))
      root.color = RED;
    root = deleteMax(root);
    if (!isEmpty()) root.color = BLACK;
    // assert check();
  }
  // delete the key-value pair with the maximum key rooted at h
  // delete the key-value pair with the maximum key rooted at h
  private Node deleteMax(Node h) { 
      if (isRed(h.left))
        h = rotateRight(h);
      if (h.right == null)
        return null;
      if (!isRed(h.right) && !isRed(h.right.left))
        h = moveRedRight(h);
      h.right = deleteMax(h.right);
      return balance(h);
    }
  /**
   * remove the specified key and its associated value from this symbol table   
   * (if the key is in this symbol table).  
   *
   * @param key the key
   * @throws NullPointerException if key is null
   */
  public void delete(Key key) {
    if (key == null) {
      throw new NullPointerException("argument to delete() is null");
    }
    if (!contains(key)) {
      return;
    }
    //if both children of root are black, set root to red
    if(!isRed(root.left) && !isRed(root.right)) {
      root.color = RED;
    }
    root = delete(root, key);
    if(!isEmpty()) {
      root.color = BLACK;
    }
  }
  // delete the key-value pair with the given key rooted at h
  // delete the key-value pair with the given key rooted at h
  private Node delete(Node h, Key key) {
    if(key.compareTo(h.key) < 0) {
      if(!isRed(h.left) && !isRed(h.left.left)) {
        h = moveRedLeft(h);
      }
      h.left = delete(h.left, key);
    }
    else {
      if(isRed(h.left)) {
        h = rotateRight(h);
      }
      if (key.compareTo(h.key) == 0 && (h.right == null)) {
        return null;
      }
      if (!isRed(h.right) && !isRed(h.right.left)) {
        h = moveRedRight(h);
      }
      if (key.compareTo(h.key) == 0) {
        Node x = min(h.right);
        h.key = x.key;
        h.val = x.val;
        h.right = deleteMin(h.right);
      }
      else {
        h.right = delete(h.right, key);
      }
    }
    return balance(h);
  }
  /***************************************************************************
  * Red-black tree helper functions.
  ***************************************************************************/
  // make a left-leaning link lean to the right
  // make a left-leaning link lean to the right
  private Node rotateRight(Node h) {
    // assert (h != null) && isRed(h.left);
    Node x = h.left;
    h.left = x.right;
    x.right = h;
    x.color = x.right.color;
    x.right.color = RED;
    x.size = h.size;
    h.size = size(h.left) + size(h.right) + 1;
    return x;
  }
  // make a right-leaning link lean to the left
  // make a right-leaning link lean to the left
  private Node rotateLeft(Node h) {
    // assert (h != null) && isRed(h.right);
    Node x = h.right;
    h.right = x.left;
    x.left = h;
    x.color = x.left.color;
    x.left.color = RED;
    x.size = h.size;
    h.size = size(h.left) + size(h.right) + 1;
    return x;
  }
  // flip the colors of a node and its two children
  // flip the colors of a node and its two children
  private void flipColors(Node h) {
    // h must have opposite color of its two children
    // assert (h != null) && (h.left != null) && (h.right != null);
    // assert (!isRed(h) && isRed(h.left) && isRed(h.right))
    //  || (isRed(h) && !isRed(h.left) && !isRed(h.right));
    h.color = !h.color;
    h.left.color = !h.left.color;
    h.right.color = !h.right.color;
  }
  // Assuming that h is red and both h.left and h.left.left
  // are black, make h.left or one of its children red.
  // Assuming that h is red and both h.left and h.left.left
  // are black, make h.left or one of its children red.
  private Node moveRedLeft(Node h) {
    // assert (h != null);
    // assert isRed(h) && !isRed(h.left) && !isRed(h.left.left);
    flipColors(h);
    if (isRed(h.right.left)) { 
      h.right = rotateRight(h.right);
      h = rotateLeft(h);
      flipColors(h);
    }
    return h;
  }
  // Assuming that h is red and both h.right and h.right.left
  // are black, make h.right or one of its children red.
  // Assuming that h is red and both h.right and h.right.left
  // are black, make h.right or one of its children red.
  private Node moveRedRight(Node h) {
    // assert (h != null);
    // assert isRed(h) && !isRed(h.right) && !isRed(h.right.left);
    flipColors(h);
    if (isRed(h.left.left)) { 
      h = rotateRight(h);
      flipColors(h);
    }
    return h;
  }
  // restore red-black tree invariant
  // restore red-black tree invariant
  private Node balance(Node h) {
    // assert (h != null);
    if (isRed(h.right)) {
      h = rotateLeft(h);
    }
    if (isRed(h.left) && isRed(h.left.left)) {
      h = rotateRight(h);
    }
    if (isRed(h.left) && isRed(h.right)) {
      flipColors(h);
    }
    h.size = size(h.left) + size(h.right) + 1;
    return h;
  }
  /***************************************************************************
   * Utility functions.
   ***************************************************************************/
   /**
   * Returns the height of the BST (for debugging).
   * @return the height of the BST (a 1-node tree has height 0)
   */
   public int height() {
     return height(root);
   }
   private int height(Node x) {
     if (x == null) {
       return -1;
     }
     return 1 + Math.max(height(x.left), height(x.right));
   }
  /***************************************************************************
   * Ordered symbol table methods.
   ***************************************************************************/
   /**
   * Returns the smallest key in the symbol table.
   * @return the smallest key in the symbol table
   * @throws NoSuchElementException if the symbol table is empty
   */
   public Key min() {
     if (isEmpty()) {
       throw new NoSuchElementException("called min() with empty symbol table");
     }
     return min(root).key;
   } 
   // the smallest key in subtree rooted at x; null if no such key
   private Node min(Node x) { 
     // assert x != null;
     if (x.left == null) {
       return x; 
     }
     else {
       return min(x.left); 
     }
   } 
   /**
   * Returns the largest key in the symbol table.
   * @return the largest key in the symbol table
   * @throws NoSuchElementException if the symbol table is empty
   */
   public Key max() {
     if (isEmpty()) {
       throw new NoSuchElementException("called max() with empty symbol table");
     }
     return max(root).key;
   } 
   // the largest key in the subtree rooted at x; null if no such key
   private Node max(Node x) { 
     // assert x != null;
     if (x.right == null) {
       return x; 
     }
     else {
       return max(x.right);     
     }
   } 
   /**
   * Returns the largest key in the symbol table less than or equal to key.
   * @param key the key
   * @return the largest key in the symbol table less than or equal to key
   * @throws NoSuchElementException if there is no such key
   * @throws NullPointerException if key is null
   */
   public Key floor(Key key) {
     if (key == null) {
       throw new NullPointerException("argument to floor() is null");
     }
     if (isEmpty()) {
       throw new NoSuchElementException("called floor() with empty symbol table");
     }
     Node x = floor(root, key);
     if (x == null) {
       return null;     
     }
     else {
       return x.key;
     }
   }  
   // the largest key in the subtree rooted at x less than or equal to the given key
   private Node floor(Node x, Key key) {
     if (x == null) {
       return null;
     }
     int cmp = key.compareTo(x.key);
     if (cmp == 0) {
       return x;
     }
     if (cmp < 0) {
       return floor(x.left, key);     
     }
     Node t = floor(x.right, key);
     if (t != null) {
       return t;     
     }
     else {
       return x;
     }
   }
   /**
   * Returns the smallest key in the symbol table greater than or equal to key.
   * @param key the key
   * @return the smallest key in the symbol table greater than or equal to key
   * @throws NoSuchElementException if there is no such key
   * @throws NullPointerException if key is null
   */
   public Key ceiling(Key key) {
     if (key == null) {
       throw new NullPointerException("argument to ceiling() is null");
     }
     if (isEmpty()) {
       throw new NoSuchElementException("called ceiling() with empty symbol table");
     }
     Node x = ceiling(root, key);
     if (x == null) {
       return null;
     }
     else {
       return x.key; 
     }
   }
   // the smallest key in the subtree rooted at x greater than or equal to the given key
   private Node ceiling(Node x, Key key) { 
     if (x == null) {
       return null;
     }     
     int cmp = key.compareTo(x.key);
     if (cmp == 0) {
       return x;
     }
     if (cmp > 0) {
       return ceiling(x.right, key);
     }
     Node t = ceiling(x.left, key);
     if (t != null) {
       return t; 
     }
     else {
       return x;
     }
   }
   /**
   * Return the kth smallest key in the symbol table.
   * @param k the order statistic
   * @return the kth smallest key in the symbol table
   * @throws IllegalArgumentException unless k is between 0 and
   *   <em>N</em> − 1
   */
   public Key select(int k) {
     if (k < 0 || k >= size()) {
       throw new IllegalArgumentException();
     }
     Node x = select(root, k);
     return x.key;
   }
   // the key of rank k in the subtree rooted at x
   private Node select(Node x, int k) {
     // assert x != null;
     // assert k >= 0 && k < size(x);
     int t = size(x.left); 
     if   (t > k) {
       return select(x.left, k); 
     }
     else if (t < k) {
       return select(x.right, k-t-1); 
     }
     else {
       return x; 
     }
   } 
   /**
   * Return the number of keys in the symbol table strictly less than key.
   * @param key the key
   * @return the number of keys in the symbol table strictly less than key
   * @throws NullPointerException if key is null
   */
   public int rank(Key key) {
     if (key == null) {
       throw new NullPointerException("argument to rank() is null");
     }
     return rank(key, root);
   } 
   // number of keys less than key in the subtree rooted at x
   private int rank(Key key, Node x) {
     if (x == null) {
       return 0; 
     }
     int cmp = key.compareTo(x.key); 
     if   (cmp < 0) {
       return rank(key, x.left); 
     }
     else if (cmp > 0) {
       return 1 + size(x.left) + rank(key, x.right); 
     }
     else {
       return size(x.left); 
     }
   } 
  /***************************************************************************
   * Range count and range search.
   ***************************************************************************/
   /**
   * Returns all keys in the symbol table as an Iterable.
   * To iterate over all of the keys in the symbol table named st,
   * use the foreach notation: for (Key key : st.keys()).
   * @return all keys in the symbol table as an Iterable
   */
   public Iterable<key> keys() {
     if (isEmpty()) {
       return new Queue<key>();
     }
     return keys(min(), max());
   }
   /**
   * Returns all keys in the symbol table in the given range,
   * as an Iterable.
   * @return all keys in the symbol table between lo 
   *  (inclusive) and hi (exclusive) as an Iterable
   * @throws NullPointerException if either lo or hi
   *  is null
   */
   public Iterable<key> keys(Key lo, Key hi) {
     if (lo == null) {
       throw new NullPointerException("first argument to keys() is null");
     }
     if (hi == null) {
       throw new NullPointerException("second argument to keys() is null");
     }
     Queue<key> queue = new Queue<key>();
     // if (isEmpty() || lo.compareTo(hi) > 0) return queue;
     keys(root, queue, lo, hi);
     return queue;
   } 
   // add the keys between lo and hi in the subtree rooted at x
   // to the queue
   private void keys(Node x, Queue<key> queue, Key lo, Key hi) { 
     if (x == null) {
       return; 
     }
     int cmplo = lo.compareTo(x.key); 
     int cmphi = hi.compareTo(x.key); 
     if (cmplo < 0) {
       keys(x.left, queue, lo, hi); 
     }
     if (cmplo <= 0 && cmphi >= 0) {
       queue.enqueue(x.key); 
     }
     if (cmphi > 0) {
       keys(x.right, queue, lo, hi); 
     }
   } 
   /**
   * Returns the number of keys in the symbol table in the given range.
   * @return the number of keys in the symbol table between lo 
   *  (inclusive) and hi (exclusive)
   * @throws NullPointerException if either lo or hi
   *  is null
   */
   public int size(Key lo, Key hi) {
     if (lo == null) {
       throw new NullPointerException("first argument to size() is null");
     }
     if (hi == null) {
       throw new NullPointerException("second argument to size() is null");
     }
     if (lo.compareTo(hi) > 0) {
       return 0;
     }
     if (contains(hi)) {
       return rank(hi) - rank(lo) + 1;
     }
     else {
       return rank(hi) - rank(lo);     
     }
   }
  /***************************************************************************
   * Check integrity of red-black tree data structure.
   ***************************************************************************/
   private boolean check() {
     if (!isBST())      System.out.println("Not in symmetric order");
     if (!isSizeConsistent()) System.out.println("Subtree counts not consistent");
     if (!isRankConsistent()) System.out.println("Ranks not consistent");
     if (!is23())       System.out.println("Not a 2-3 tree");
     if (!isBalanced())    System.out.println("Not balanced");
     return isBST() && isSizeConsistent() && isRankConsistent() && is23() && isBalanced();
   }
   // does this binary tree satisfy symmetric order?
   // Note: this test also ensures that data structure is a binary tree since order is strict
   private boolean isBST() {
     return isBST(root, null, null);
   }
   // is the tree rooted at x a BST with all keys strictly between min and max
   // (if min or max is null, treat as empty constraint)
   // Credit: Bob Dondero's elegant solution
   private boolean isBST(Node x, Key min, Key max) {
     if (x == null) {
       return true;
     }
     if (min != null && x.key.compareTo(min) <= 0) {
       return false;     
     }
     if (max != null && x.key.compareTo(max) >= 0) {
       return false;
     }
     return isBST(x.left, min, x.key) && isBST(x.right, x.key, max);
   } 
   // are the size fields correct?
   private boolean isSizeConsistent() { 
     return isSizeConsistent(root); 
   }
   private boolean isSizeConsistent(Node x) {
     if (x == null) {
       return true;
     }
     if (x.size != size(x.left) + size(x.right) + 1) {
       return false;
     }
     return isSizeConsistent(x.left) && isSizeConsistent(x.right);
   } 
   // check that ranks are consistent
   private boolean isRankConsistent() {
     for (int i = 0; i < size(); i++) {
       if (i != rank(select(i))) {
         return false;
       }
     }
     for (Key key : keys()) {
       if (key.compareTo(select(rank(key))) != 0) {
         return false;
       }
     }
     return true;
   }
   // Does the tree have no red right links, and at most one (left)
   // red links in a row on any path?
   private boolean is23() { 
     return is23(root); 
   }
   private boolean is23(Node x) {
     if (x == null) {
       return true;
     }
     if (isRed(x.right)) {
       return false;
     }
     if (x != root && isRed(x) && isRed(x.left)){
       return false;
     }
     return is23(x.left) && is23(x.right);
   } 
   // do all paths from root to leaf have same number of black edges?
   private boolean isBalanced() { 
     int black = 0;   // number of black links on path from root to min
     Node x = root;
     while (x != null) {
       if (!isRed(x)) black++;
       x = x.left;
     }
     return isBalanced(root, black);
   }
   // does every path from the root to a leaf have the given number of black links?
   private boolean isBalanced(Node x, int black) {
     if (x == null) {
       return black == 0;     
     }
     if (!isRed(x)) {
       black--;
     }
     return isBalanced(x.left, black) && isBalanced(x.right, black);
   } 
   /**
   * Unit tests the RedBlackBST data type.
   */
   public static void main(String[] args) { 
     RedBlackBST<string, integer=""> st = new RedBlackBST<string, integer="">();
     String data = "a b c d e f g h m n o p";
     Scanner sc = new Scanner(data);
     int i = 0;
     while (sc.hasNext()) {     
      String key = sc.next();
      st.put(key, i);
      i++;
     }
     sc.close();   
     for (String s : st.keys())
       System.out.println(s + " " + st.get(s));
     System.out.println();
     boolean result = st.check();
     System.out.println("check: " + result);
   }
 }

输出:

<code>a 0
b 1
c 2
d 3
e 4
f 5
g 6
h 7
m 8
n 9
o 10
p 11
 
check: true</code>

总结

以上就是本文关于java算法实现红黑树完整代码示例的全部内容,希望对大家有所帮助。感兴趣的朋友可以继续参阅本站:java集合中list的用法代码示例、Java微信支付之服务号支付代码示例、快速理解Java设计模式中的组合模式等,有什么问题可以随时留言,小编会及时回复大家的。感谢朋友们对本站的支持!

 类似资料:
  • 本文向大家介绍Java编程实现A*算法完整代码,包括了Java编程实现A*算法完整代码的使用技巧和注意事项,需要的朋友参考一下 前言 A*搜寻算法俗称A星算法。这是一种在图形平面上,有多个节点的路径,求出最低通过成本的算法。常用于游戏中 通过二维数组构建的一个迷宫,“%”表示墙壁,A为起点,B为终点,“#”代表障碍物,“*”代表算法计算后的路径 本文实例代码结构: 算法理论 算法的核心公式为:F=

  • 我知道Dijkstra的算法实际上是使用斐波那契堆实现的。但是,它也可以使用红色的黑树来实现,并且仍然具有O(m log n)的最坏情况运行时间吗?

  • 本文向大家介绍红黑树的插入详解及Javascript实现方法示例,包括了红黑树的插入详解及Javascript实现方法示例的使用技巧和注意事项,需要的朋友参考一下 红黑树的性质 一棵满足以下性质的二叉搜索树是一棵红黑树 每个结点或是黑色或是红色。 根结点是黑色的。 每个叶结点(NIL)是黑色的。 如果一个结点是红色的,则它的两个子结点都是黑色的。 对每个结点,从该结点到其所有后代叶结点的简单路径上

  • 本文向大家介绍Java中filter用法完整代码示例,包括了Java中filter用法完整代码示例的使用技巧和注意事项,需要的朋友参考一下 本文研究的主要是Java中filter过滤器的相关用法,具体实现代码如下。 filter过滤器主要使用于前台向后台传递数据是的过滤操作。程度很简单就不说明了,直接给几个已经写好的代码: 一、使浏览器不缓存页面的过滤器 二、检测用户是否登陆的过滤器 三、字符编码

  • 红黑树与AVL的比较: AVL是严格平衡树,因此在增加或者删除节点的时候,根据不同情况,旋转的次数比红黑树要多; 红黑是用非严格的平衡来换取增删节点时候旋转次数的降低; 所以简单说,如果你的应用中,搜索的次数远远大于插入和删除,那么选择AVL,如果搜索,插入删除次数几乎差不多,应该选择RB。 红黑树详解: https://xieguanglei.github.io/blog/post/red-bl

  • 本文向大家介绍图解红黑树及Java进行红黑二叉树遍历的方法,包括了图解红黑树及Java进行红黑二叉树遍历的方法的使用技巧和注意事项,需要的朋友参考一下 红黑树 红黑树是一种数据结构与算法课堂上常常提到但又不会细讲的树,也是技术面试中经常被问到的树,然而无论是书上还是网上的资料,通常都比较刻板难以理解,能不能一种比较直观的方式来理解红黑树呢?本文将以图形的方式来解释红黑树的插入与删除操作。 对树结构