加入收藏 | 设为首页 | 会员中心 | 我要投稿 济南站长网 (https://www.0531zz.com/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 教程 > 正文

二叉搜索树之Java达成

发布时间:2021-11-20 17:07:32 所属栏目:教程 来源:互联网
导读:什么是二叉搜索树 二叉搜索树(Binary Search Tree),是最基础,且相对简单的一种数据结构,支持Insert,Delete,Search,Min,Max,Successor,Predecessor等操作。最大的特点是每一个节点有不超过两个子节点,并且左子节点小于或者等于父节点,而右节点大于

什么是二叉搜索树
 
二叉搜索树(Binary Search Tree),是最基础,且相对简单的一种数据结构,支持Insert,Delete,Search,Min,Max,Successor,Predecessor等操作。最大的特点是每一个节点有不超过两个子节点,并且左子节点小于或者等于父节点,而右节点大于或者等于父节点。说它基础,是因为很多其它树形数据结构以它为原型而扩展,比如红黑树,B树。
 
具体实现
 
public class BinaryTree<T extends Comparable<T>> {
 private Node<T> root;
 
 public void insert(T element) {
  if (element == null) {
   throw new IllegalArgumentException("element can not be null");
  }
 
  if (root == null) {
   root = new Node<T>(null, element);
  } else {
   Node<T> node = root;
   while (true) {
    if (element.compareTo(node.value) <= 0) {
     if (node.left == null) {
      Node<T> newNode = new Node<T>(node, element);
      node.left = newNode;
      break;
     } else {
      node = node.left;
     }
    } else {
     if (node.right == null) {
      Node<T> newNode = new Node<T>(node, element);
      node.right = newNode;
      break;
     } else {
      node = node.right;
     }
    }
   }
  }
 }
 
 private int childCount(Node<T> node) {
  if (node == null) {
   throw new IllegalArgumentException("node can not be null");
  }
 
  int count = 0;
 
  if (node.left != null) {
   count++;
  }
 
  if (node.right != null) {
   count++;
  }
 
  return count;
 }
 
 public void delete(Node<T> node) {
  if (node == null) {
   throw new IllegalArgumentException("node can not be null");
  }
 
  int childCount = childCount(node);
  Node<T> parentNode = node.parent;
 
  if (childCount == 0) {
   if (parentNode == null) {
    // node is root
    root = null;
   } else {
    if (node == parentNode.left) {
     parentNode.left = null;
    } else {
     parentNode.right = null;
    }
   }
  } else if (childCount == 1) {
   if (parentNode == null) {
    // node is root, set child of node to be new root
    if (node.left != null) {
     root = node.left;
     node.left.parent = null;
    } else {
     root = node.right;
     node.right.parent = null;
    }
   } else {
    if (node == parentNode.left) {
     if (node.left != null) {
      parentNode.left = node.left;
      node.left.parent = parentNode;
     } else {
      parentNode.left = node.right;
      node.right.parent = parentNode;
     }
    } else {
     if (node.left != null) {
      parentNode.right = node.left;
      node.left.parent = parentNode;
     } else {
      parentNode.right = node.right;
      node.right.parent = parentNode;
     }
    }
   }
  } else {
   // successor has no left child
   Node<T> successor = min(node);
 
   if (successor != node.right) {
    transplant(successor, successor.right);
 
    successor.right = node.right;
    node.right.parent = successor;
   }
 
   transplant(node, successor);
 
   successor.left = node.left;
   node.left.parent = successor;
  }
 }
 
 private void transplant(Node<T> u, Node<T> v) {
  if (u == null) {
   throw new IllegalArgumentException("node can not be null");
  }
 
  if (u.parent == null) {
   root = v;
  } else if (u == u.parent.left) {
   u.parent.left = v;
  } else {
   u.parent.right = v;
  }
 
  if (v != null) {
   v.parent = u.parent;
  }
 }
 
 public Node<T> search(T element) {
  if (element == null) {
   throw new IllegalArgumentException("element can not be null");
  }
 
  Node<T> node = root;
  while (node != null) {
   if (node.value.equals(element)) {
    return node;
   } else if (element.compareTo(node.value) < 0) {
    node = node.left;
   } else {
    node = node.right;
   }
  }
 
  return null;
 }
 
 public Node<T> min(Node<T> rootNode) {
  if (rootNode == null) {
   throw new IllegalArgumentException("node can not be null");
  }
 
  Node<T> node = rootNode;
  while (node.left != null) {
   node = node.left;
  }
 
  return node;
 }
 
 public Node<T> min() {
  if (root != null) {
   return min(root);
  } else {
   return null;
  }
 }
 
 public Node<T> max(Node<T> rootNode) {
  if (rootNode == null) {
   throw new IllegalArgumentException("node can not be null");
  }
 
  Node<T> node = rootNode;
  while (node.right != null) {
   node = node.right;
  }
 
  return node;
 }
 
 public Node<T> max() {
  if (root != null) {
   return max(root);
  } else {
   return null;
  }
 }
 
 public Node<T> successor(Node<T> node) {
  if (node == null) {
   throw new IllegalArgumentException("node can not be null");
  }
 
  if (node.right != null) {
   return min(node.right);
  }
 
  Node<T> processNode = node;
  Node<T> parent = processNode.parent;
  while (parent != null && processNode == parent.right) {
   processNode = parent;
   parent = processNode.parent;
  }
 
  return parent;
 }
 
 public Node<T> predecesssor(Node<T> node) {
  if (node == null) {
   throw new IllegalArgumentException("node can not be null");
  }
 
  if (node.left != null) {
   return max(node.left);
  }
 
  Node<T> processNode = node;
  Node<T> parent = processNode.parent;
  while (parent != null && processNode == parent.left) {
   processNode = parent;
   parent = processNode.parent;
  }
 
  return parent;
 }
 
 public void print() {
  print(root);
 }
 
 public void print(Node<T> node) {
  if (node == null) {
   return;
  }
 
  print(node.left);
  System.out.print("  " + node.value.toString() + "  ");
  print(node.right);
 }
 
 public static class Node<T extends Comparable<T>> {
  private Node<T> parent;
  private Node<T> left;
  private Node<T> right;
 
  private T value;
 
  public Node(Node<T> parent, T value) {
   this.parent = parent;
   this.value = value;
  }
 
  public Node<T> getParent() {
   return parent;
  }
 
  public void setParent(Node<T> parent) {
   this.parent = parent;
  }
 
  public Node<T> getLeft() {
   return left;
  }
 
  public void setLeft(Node<T> left) {
   this.left = left;
  }
 
  public Node<T> getRight() {
   return right;
  }
 
  public void setRight(Node<T> right) {
   this.right = right;
  }
 
  public T getValue() {
   return value;
  }
 
  public void setValue(T value) {
   this.value = value;
  }
 }
 
 public static void main(String[] args) {
  BinaryTree<String> tree = new BinaryTree<String>();
 
  tree.insert("Hello");
  tree.insert("World");
  tree.insert("Money");
 
  tree.print();
  System.out.println();
 
  Node<String> moneyNode = tree.search("Money");
  tree.print(moneyNode);
  System.out.println();
 
  tree.insert("Like");
  tree.print(moneyNode);
  System.out.println();
 
  tree.delete(moneyNode);
  tree.print();
  System.out.println();
 }
}

(编辑:济南站长网)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    热点阅读