二叉搜索 java_数据结构之二叉搜索树

本文链接：https://blog.csdn.net/weixin_31508009/article/details/114939291

二叉查找树(Binary Search Tree，简称BST)是一棵二叉树，它的左子节点的值比父节点的值要小，右节点的值要比父节点的值大。它的高度决定了它的查找效率。

在理想的情况下，二叉查找树增删查改的时间复杂度为O(logN)(其中N为节点数)，最坏的情况下为O(N)。当它的高度为logN+1时，我们就说二叉查找树是平衡的。

首先定义二叉树节点

class TreeNode{

TreeNode(int value) {

this.value = value;

}

TreeNode left;

TreeNode right;

int value;

}

复制代码

本文中rootNode为创建二叉搜索树的root节点,在类初始化阶段赋值。

private final TreeNode rootNode;

public BinarySearchTree(int value){

rootNode = new TreeNode(value);

}

复制代码

插入操作:

首先从root节点开始循环遍历,

如果插入节点等于当前root节点,直接结束不做任何处理。

如果插入节点小于当前root节点,从左节点开始查找。 (root=root.left)

如果插入节点大于当前root节点,从右节点开始查找。 (root=root.right)

当当前节点的左节点或者右节点为空时，插入新节点，结束过程。

上代码：

public TreeNode insert(int value){

TreeNode currentRoot = rootNode;

TreeNode newNode = new TreeNode(value);

while (true) {

if (value == currentRoot.value) {

return null;

}

else if (value < currentRoot.value) {

if (currentRoot.left != null) {

currentRoot = currentRoot.left;

}

else {

return currentRoot.left = newNode;

}

else {

if (currentRoot.right != null) {

currentRoot = currentRoot.right;

}

else {

return currentRoot.right = newNode;

}

复制代码

查找操作:

类似于插入操作：

如果插入节点等于root节点,查找完成。

如果插入节点小于root节点,从left节点开始查找。 (root=root.left)

如果插入节点大于root节点,从right节点开始查找。(root=root.right)

上代码：

public TreeNode get(int value){

TreeNode currentRoot = rootNode;

while (true) {

if (currentRoot == null) {

return null;

}

if (value == currentRoot.value) {

return currentRoot;

}

else if (value < currentRoot.value) {

currentRoot = currentRoot.left;

}

else {

currentRoot = currentRoot.right;

}

复制代码

遍历操作:

遍历分为

前序：根节点放在左节点的左边 (根→左→右)

中序：根节点放在左节点和右节点的中间(左→根→右)

后序：根节点放在右节点的右边 (左→右→根)

中序遍历的代码：

public void getTreeByInOrder(List list, TreeNode root){

if (root != null) {

getTreeByInOrder(list, root.left);

list.add(root);

getTreeByInOrder(list, root.right);

}

复制代码

前序后序和其类似，就不贴代码了

层次遍历(也可以叫做广度优先遍历)：

思路:

由根从上往下循环遍历，利用队列(queue)先进先出的原则，每访问一个节点之后，将其左右子节点入队。这样，同一层的所有节点肯定先于下一层的节点先访问

如果需要分层保存节点呢?(比如需要换行打印)：问题的难点是如何知道本层已经遍历完了，可以使用两个指针last和nextLast，一个指向当前层的最后一个节点，一个指向下一层的最后一个节点，这样难点转换成了last和nextLast的更新。当节点入队时，更新nextLast(设置nextLast为队列的最后一个节点)，因为队列中的最后一个节点一定是下一层的最后节点。

上代码：

public List> printTreeByHierarchy() {

TreeNode currentRoot = rootNode;

Queue queue = new LinkedList<>();

TreeNode last = currentRoot;

TreeNode nextLast = null;

queue.offer(currentRoot);

List> lists = new ArrayList<>();

List nodes = new ArrayList<>();

while (!queue.isEmpty()) {

currentRoot = queue.poll();

nodes.add(currentRoot.value);