查找二叉树(插入、删除、查找)实现

二叉查找树(Binary Search Tree),(又:二叉搜索树,二叉排序树)它或者是一棵空树,或者是具有下列性质的二叉树: 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值; 它的左、右子树也分别为二叉排序树

 

如图:

基于这样的特性,查找的时候就很好操作了,从根节点开始,查找,如果值大于节点值,往右找;如果值小于节点值,往左找;如果值刚好相等,就找到了。是不是看着就能写出代码了?这种查找过程很像二分查找法,但是那个是数组结构,这个是树结构。

      二叉查找树的操作基本概括为:插入值,删除值,查找值以及二叉树的遍历。

      注意的是:这里面,删除是最麻烦的。

      (本来觉得写数据结构还是用c语言最好的,直接可以操作指针,清晰明了效率高,但是c确实丢了太久了,而且现在主要目的是温习数据结构和算法的知识,所以只能放弃用c的想法,以后如果需要再学习,先用最熟悉的java来实现代码)

      下面来看具体的操作和逻辑,附带贴上代码。

 

我们直接说最麻烦的删除节点吧!!

首先我们要删除一个节点,而这个节点他可能有左子节点或者右子节点或者没有子节点。。就比如图中

  (1)我要删除55这个节点,我们可以直接删除他,中序遍历的结果还是很正常

  (2)我要删除53这个节点,这是个有左无右的节点,那我们可以直接让被他的左子节点取代

  (3)我要删除67或者98这种节点,因为他们都有右子树,我们就要找到右子树的最小值来作为前继节点(比要删除的节点大的节点里最小的)用来取代当前node的值,并且把前继节点从原先位置删除!

 

具体实现代码如下所示:

public class SearchTree {
	private TreeNode root;

	public SearchTree() {
	}

	// 中序遍历(递归)
	public void midOrder(TreeNode node) {
		if (node == null) {
			return;
		}
		// 对该节点继续进行左右子节点的遍历
		// 先输出根节点,因为这是前序遍历,根左右
		midOrder(node.lchild);
		System.out.println("二叉树节点 : " + node.data);
		midOrder(node.rchild);
	}

	public TreeNode putTreeNode(int key) {
		TreeNode node = null;
		TreeNode parent = null;
		if (root == null) {
			node = new TreeNode(0, key);
			root = node;
		}
		// 如果根节点已经存在,就要开始判断了
		node = root;
		while (node != null) {
			parent = node;
			if (key > node.data) {
				// 获取右节点
				node = node.rchild;
			} else if (key < node.data) {
				// 获取左节点
				node = node.lchild;
			} else {
				// 相等的话,就啥也不做
				return node;
			}
		}
		// 如果跳出了循环,代表这个节点不存在,需要创建
		node = new TreeNode(0, key);
		// 找到此时node的父节点
		if (key > parent.data) {
			parent.rchild = node;
		} else if (key < parent.data) {
			parent.lchild = node;
		}
		node.parent = parent;
		return node;
	}

	// 删除节点
	public void deleteNode(int key) {
		TreeNode node = searchNode(key);
		if (node == null) {
			throw new RuntimeException("查找该节点不存在!!");
		}
		delete(node);
	}

	public void delete(TreeNode node) {
		if (node == null) {
			throw new RuntimeException("删除该节点不存在!!");
		}
		TreeNode parent = node.parent;
		if (node.lchild == null && node.rchild == null) {
			// 直接删除,不要顾及太多
			if (parent.lchild == node) {
				parent.lchild = null;
			}
			if (parent.rchild == node) {
				parent.rchild = null;
			}
			return;
		}
		// 如果是有左无右的,左节点默认是要删除节点的后继节点
		if (node.lchild != null && node.rchild == null) {
			if (parent.lchild == node) {
				parent.lchild = node.lchild;
			}
			if (parent.rchild == node) {
				parent.rchild = node.lchild;
			}
			return;
		}
		// 如果是有右无左或者是有右有左,我们都选择拿到右子树的最小值当前继节点
		// 我们首先获取到前继节点
		TreeNode next = getNextNode(node);
		// 我们要确定一点的是,我们拿到的前继节点一定是没有左子节点的
		// 拿到前继节点后,我们要删除这个节点,
		// 就是说要把前继节点的右子树设置为前继节点的父节点的左子树
		delete(next);
		// 把前继节点的值赋值给要删除的node
		node.data = next.data;
	}

	/**
	 * @param node
	 *            查找后继节点
	 * @return
	 */
	public TreeNode getNextNode(TreeNode node) {
		if (node == null) {
			return null;
		}
		// 其实既可以找左子树最大的那个值也可以找右子树最小的值
		if (node.rchild != null) {
			// 这里选择的是拿右子树最小的值
			// 找某节点最小关键字节点(右子树最小值)
			return getMinTreeNode(node.rchild);
		}
		return null;
	}

	private TreeNode getMinTreeNode(TreeNode node) {
		// 一直找左孩子
		if (node == null) {
			return null;
		}
		TreeNode parent = null;
		while (node != null) {
			parent = node;
			node = node.lchild;
		}
		return parent;
	}

	// 查找节点
	public TreeNode searchNode(int key) {
		if (root == null) {
			return null;
		}
		TreeNode node = root;
		while (node != null) {
			if (key > node.data) {
				node = node.rchild;
			} else if (key < node.data) {
				node = node.lchild;
			} else {
				return node;
			}
		}
		// 跳出了上面循环的话,就代表没有该节点。。。
		return null;
	}

	public class TreeNode {
		private int index;
		private int data;
		private TreeNode lchild;
		private TreeNode rchild;
		private TreeNode parent;

		public TreeNode(int index, int data) {
			this.index = index;
			this.data = data;
		}

	}

	public static void main(String[] args) {
		SearchTree searchTree = new SearchTree();
		int[] array = { 50, 30, 15, 45, 60, 55, 70, 58 };
		for (int data : array) {
			searchTree.putTreeNode(data);
		}
		searchTree.midOrder(searchTree.root);
		System.out.println("============");
		searchTree.deleteNode(60);
		searchTree.midOrder(searchTree.root);
	}
}

  

posted @ 2018-04-22 22:22  Booker808  阅读(586)  评论(0编辑  收藏  举报