输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点,只能调整树中节点指针的指向。
为了让您更好地理解问题,以下面的二叉搜索树为例:
我们希望将这个二叉搜索树转化为双向循环链表。链表中的每个节点都有一个前驱和后继指针。对于双向循环链表,第一个节点的前驱是最后一个节点,最后一个节点的后继是第一个节点。
下图展示了上面的二叉搜索树转化成的链表。“head” 表示指向链表中有最小元素的节点。
特别地,我们希望可以就地完成转换操作。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继。还需要返回链表中的第一个节点的指针。
"""
# Definition for a Node.
class Node:
def __init__(self, val, left=None, right=None):
self.val = val
self.left = left
self.right = right
"""
class Solution:
def treeToDoublyList(self, root: 'Node') -> 'Node':
self.head = None
self.pre = None
self.cur = None
# 一共三个关键节点,head:头节点,cur:当前节点,pre:上一个节点
# 首先使用的是中序遍历,因为二叉搜索树,左子树的值一定的都小于根节点的
# 右子树的值一定都大于根节点
# 而中序遍历的顺序是左,中,右的顺序进行遍历的,故中序遍历后的结果一定是从小到大的
def dfs(root):
# 如果节点是None,则直接返回,递归的终止条件
if not root:
return
# 左子树遍历
dfs(root.left)
# 第一次是最左边左叶子节点,该值是最小值
if self.head == None:
# 最小值是head节点
self.head = root
# 当前节点的前节点
self.pre = root
# 当前节点,头节点,当前节点根前节点都是头节点
self.cur = root
else:
# 非头节点,当前直接向后移动
self.cur = root
# 因为循环双向链表,是从小到大排序,那么.right右节点是指向当前节点的,当前节点的值大于前节点的值
self.pre.right = self.cur
# 同理,当前节点的左节点是前一个节点
self.cur.left = self.pre
# 前移向当前节点
self.pre = self.cur
# print(root.val)
# 右节点遍历
dfs(root.right)
# 最后的遍历结束后,cur是尾节点,头节点的左节点指向尾节点
self.head.left = self.cur
# 尾节点的右节点指向头节点,这样就构成了循环双向链表
self.cur.right = self.head
dfs(root)
return self.head