二叉搜索树

本文参考：《算法导论》P286~P307
注：每种操作下列举了一些leetcode上的相关习题，可以用作练习。

二叉搜索树

  二叉搜索树由一棵二叉树组织。这样的一棵树可以使用一个链表的数据结构来表示，每个节点就是一个对象。节点的属性值包含$left$（左孩子）、$right$（右孩子）、$parent$（双亲）、$key$（关键字/节点值） ，如果二叉搜索树的某个节点的孩子节点或双亲节点不存在，其对应的属性值就为$None$。（根节点是二叉搜索树中唯一父指针为$None$的节点）
  二叉搜索树支持的操作有：$SEARCH$（查找）、$MINIMUM$（最小值）、$MAXIMUM$（最大值）、$SUCCESSOR$（后继）、$PREDECESSOR$（前驱）等。

1 二叉搜索树的性质

  性质：对于任何节点$x$，其左子树中的关键字最大不超过$x.key$，其右子树中的关键字最小不小于$x.key$。不同的二叉搜索树可以代表同一组值的集合。大部分搜索树的操作的最坏运行时间与树的高度成正比。
  二叉搜索树的关键字$key$总是以满足二叉搜索树性质的方式来存储，设$x$是二叉搜索树的一个节点，则：

• 如果$y$是$x$左子树中的一个节点，那么$y.key\le x.key$。
• 如果$y$是$x$右子树中的一个节点，那么$y.key\ge x.key$。

  二叉搜索树的性质允许我们通过中序遍历（$inordertreewalk$）来输出二叉搜索树中的所有关键字。leetcode 94.二叉树的中序遍历可以用作练习。下面是中序遍历的递归版伪代码描述。（非递归需要借助栈的数据结构）

$INORDER-TREE-WALK(T.root)$：中序遍历二叉树$T$。

INORDER-TREE-WALK(x):
if x != None:
	INORDER-TREE-WALK(x.left)
	print x.key
	INORDER-TREE-WALK(x.right)
end if

与中序遍历有关的二叉搜索树基本性质的练习还有：

• Leetcode 98.验证二叉搜索树
• Leetcode 99.恢复二叉搜索树
• Leetcode 173.二叉搜索树迭代器
• Leetcode 230.二叉搜索树中第K小的元素
• Leetcode 530.二叉搜索树的最小绝对差
• Leetcode 538.把二叉搜索树转换为累加树
• Leetcode 783.二叉搜索树结点最小距离
• Leetcode 938.二叉搜索树的范围和

2 查询二叉搜索树

  我们经常需要查找一个存储在二叉搜索树中的关键字。这些查询操作在高度为$h$的二叉搜索树上，都可以在$O(h)$的时间内执行完毕。

2.1 查找（$SEARCH$）

  输入一个指向树根的指针和一个关键字$k$，如果这个节点存在，$TREE-SEARCH$返回一个指向关键字为$k$的节点的指针，否则返回$None$。算法伪代码如下：

$TREE-SEARCH(x,k)$：找到关键字为$k$的节点。（递归版本）

TREE-SEARCH(x,k):
if x == None or k == x.key:
	return x
end if
if k < x.key:
	return TREE-SEARCH(x.left, k)
else:
	return TREE-SEARCH(x.right, k)
end if 

$ITERATIVE-TREE-SEARCH(x,k)$：找到关键字为$k$的节点。（迭代版本）

ITERATIVE-TREE-SEARCH(x,k):
while x != None and k != x.key:
	if k < x.key:
		x = x.left
	else:
		x = x.right
	end if
end while
return x

与查找有关的练习：

• Leetcode 700.二叉搜索树中的搜索

2.2 最大/最小关键字（$MAXIMUM,MINMUM$）

  根据二叉搜索树的性质，我们可以很容易地想到，整棵树中，一直向左的最后一个孩子的关键字就是最小的。相对地，一直向右的最后一个孩子的关键字就是最大的。算法的伪代码描述如下：

$TREE-MINIMUM(x)$，返回二叉搜索树关键字最小的节点。

TREE-MINIMUM(x):
while x.left != None:
	x = x.left
end while
return x

$TREE-MAXIMUM(x)$，返回二叉搜索树关键字最大的节点。

TREE-MAXIMUM(x):
while x.right != None:
	x = x.right
end while
return x

2.3 后继和前驱（$SUCCESSOR,PREDECESSOR$）

  给定二叉搜索树中的一个节点，有时候需要按中序遍历的次序查找它的后继。如果所有的关键字互不相同，则一个节点$x$的后继是大于$x.key$的最小关键字的节点。一棵二叉搜索树的结构允许我们通过没有任何关键字的比较来确定一个节点的后继。下面用伪代码来描述查找节点$x$的后继：

$TREE-SUCCESSOR(x)$，寻找节点$x$的后继。

TREE-SUCCESSOR(x):
if x.right != None:
	return TREE-MINIMUM(x.right)
end if
y = x.parent
while y != None and x == y.right:
	x = y
	y = y.parent
end while
return y

根据伪代码，可以看出$TREE-SUCCESSOR$被分成了两个部分：

• 如果节点$x$的右子树非空，那么$x$的后继就是$x$右子树中的最左节点$TREE-MINIMUM(x.right)$
• 如果节点$x$的右子树为空并且$x$有一个后继$y$，那么$y$就是$x$的最底层祖先，并且$y$的左孩子也是$x$的一个祖先。

要理解这两种情况只需要根据伪代码思考一下中序遍历的过程。

  节点$x$的前驱的查找与后继刚好是对称的，思考一下如何逆序输出一棵二叉搜索树，答案就是将中序遍历倒过来。书上并没有给出前驱的伪代码，这里做一下补充。伪代码如下：

$TREE-PREDECESSOR(x)$，寻找节点$x$的前驱。

TREE-PREDECESSOR(x)：
if x.left != None:
	return TREE-MAXIMUM(x.left)
end if
y = x.parent
while y != None and x == y.left:
	x = y
	y = y.parent
end while
return y

2.4 插入（$INSERT$）

  插入操作会引起由二叉搜索树表示的动态集合的变化。一定要修改数据结构来反映这个变化，同时修改不能破坏二叉搜索树的性质。
  将一个新的值$v$插入到一个二叉搜索树$T$中，需要调用过程$TREE-INSERT$。该过程以节点$z$作为输入，其中$z.key=v,z.left=None,z.right=None$。算法的伪代码描述如下：

$TREE-INSERT(T,z)$，将值为$v$的节点$z$插入树$T$。

TREE-INSERT(T,z):
y = None
x = T.root
while x != None:
	y = x
	if z.key < x.key:
		x = x.left
	else:
		x = x.right
	end if
end while
z.p = y
if y == None:
	T.root = z   //树T为空树
else if z.key < y.key:
	y.left = z
else:
	y.right = z
end if

那么分析一下这个伪代码：

1. 首先$y$一直记录$x$的父节点，代码的$while$循环部分类似于$SEARCH$操作，找到$z$应该插入的位置，最后$x$为空的位置就是$z$要插入的位置。
2. 接下来判断了一下$y$是否为空，如果为空说明是一棵空树。
3. 最后判断$z$应该是$y$的左孩子还是右孩子。

二叉搜索树插入的练习：

• Leetcode 701.二叉搜索树中的插入操作

2.5 删除（$DELETE$）

  删除操作与插入操作同样需要修改数据结构，从一棵二叉搜索树$T$中删除一个节点$z$的整个策略分为三种基本情况：

• 如果$z$没有孩子节点，那么只是简单地将它删除，并修改它的父节点，用$None$作为孩子来替换$z$。
• 如果$z$只有一个孩子，那么将这个孩子提升到树中$z$的位置上，并修改$z$的父节点，用$z$的孩子来替换$z$。
• 如果$z$有两个孩子，那么找$z$的后继$y$（一定在$z$的右子树中），并让$y$占据树中$z$的位置。$z$的原来右子树部分称为$y$ 的新的右子树，并且$z$的左子树成为$y$的新的左子树。

前两种情况都很简单，第三种情况比较麻烦，因为还与$y$是否为$z$的右孩子相关。那么，结合图来具体分析一下上面三种情况。

（1）如果$z$没有左孩子，那么用其右孩子来替换$z$，这个右孩子可以是$None$，也可以不是。如果是$None$则对应于上述第一种情况，如果不是$None$则对应于上述第二种情况，如下图所示。

（2）如果$z$只有一个孩子且为其左孩子，那么用左孩子来替换$z$，如下图所示。

（3）如果$z$有两个孩子，那么就要查找$z$的后继$y$，这个后继位于$z$的右子树中，并且没有左孩子。需要将$y$移出原来的位置进行拼接，并替换树中的$z$。这时，如果$y$是$z$的右孩子，那么用$y$替换$z$，并仅留下$y$的右孩子，如下图所示。

（4）接着（3）,$y$位于$z$的右子树中但不是$z$的右孩子。在这种情况下，先用$y$的右孩子替换$y$，然后再用$y$替换$z$，如下图所示。

  为了在二叉搜索树中内移动子树，定义一个子过程$TRANSPLANT$，它是用另一棵子树替换一棵子树并成为其双亲的孩子节点。当$TRANSPLANT$用一棵以$v$为根的子树来替换一棵以$u$为根的子树时，节点$u$的双亲就变为节点$v$的双亲，并且最后$v$成为$u$的双亲的相应孩子。算法伪代码描述如下：

$TRANSPLANT(T,u,v)$，用$v$替换$u$。

TRANSPLANT(T,u,v):
if u.parent == None:   //处理u为根节点的情况
	T.root = v
else if u == u.parent.left:
	u.parent.left = v
else:
	u.parent.right = v
end if
if v != None:		//允许v为None
	v.parent = u.parent  
end if

那么通过$TRANSPLANT$以及查找节点后继的方法，我们就可以写出从二叉搜索树$T$中删除节点$z$的删除过程，伪代码描述如下：

$TREE-DELETE(T,z)$，从二叉搜索树$T$中删除节点$z$。

TREE-DELETE(T,z):
if z.left == None:
	TRANSPLANT(T, z, z.right)
else if z.right = None:
	TRANSPLANT(T, z, z.left)
else:
	y = TREE-MINIMUM(z.right)
	if y.parent != z:
		TRANSPLANT(T, y, y.right)
		y.right = z.right
		y.right.parent = y
	end if
	TRANSPLANT(T, z, y)
	y.left = z.left
	y.left.parent = z
end if

二叉搜索树节点删除的相关练习：

• Leetcode 450.删除二叉搜索树中的节点