题目
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点,只能调整树中节点指针的指向。
为了让您更好地理解问题,以下面的二叉搜索树为例:
我们希望将这个二叉搜索树转化为双向循环链表。链表中的每个节点都有一个前驱和后继指针。对于双向循环链表,第一个节点的前驱是最后一个节点,最后一个节点的后继是第一个节点。
下图展示了上面的二叉搜索树转化成的链表。“head” 表示指向链表中有最小元素的节点。
特别地,我们希望可以就地完成转换操作。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继。还需要返回链表中的第一个节点的指针。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-yu-shuang-xiang-lian-biao-lcof
解题思路
二叉搜索树:
- 左子节点的值 < 根节点的值
- 右子节点的值 > 根节点的值
要求不能创建新的节点:考虑树的遍历
排序的循环双向链表:二叉搜索树的中序遍历就是升序排序 - 左根右
- 维护两个指针:头指针head和前驱节点指针pre
- 对树进行中序遍历,当前节点记为cur
- 当pre == null时,说明当前节点的前驱节点是null,此时当前节点就是二叉搜索树中最小的节点,也就是需要返回的循环链表的头节点,head = cur
- 当pre != null时,说明当前节点有前驱节点,此时pre.right = cur
- 当前节点指向前驱节点,这一步并不需要判断前驱节点是否为null,cur.left = pre
- 前驱节点后移,pre = cur
- 结束中序遍历后
- 头节点的前驱节点指向尾节点
- 尾节点的后继节点指向头节点
- 返回头节点
代码
/*
// Definition for a Node.
class Node {
public int val;
public Node left;
public Node right;
public Node() {}
public Node(int _val) {
val = _val;
}
public Node(int _val,Node _left,Node _right) {
val = _val;
left = _left;
right = _right;
}
};
*/
class Solution {
// 返回的头节点
private Node head;
// 当前节点的前驱节点
private Node pre;
public Node treeToDoublyList(Node root) {
// 特殊情况
if (root == null) {
return null;
}
// 中序遍历
dfs(root);
// 头节点指向尾节点
head.left = pre;
// 尾节点指向头节点
pre.right = head;
return head;
}
private void dfs(Node cur) {
// 递归终止条件
if (cur == null) {
return;
}
// 进行中序遍历,左根右
dfs(cur.left);
// 当pre为null时,说明此时的cur就是这个二叉树中最小的元素,也就是需要返回的链表的头节点
if (pre == null) {
head = cur;
} else {
// 不为null,则右子树指向当前节点
pre.right = cur;
}
// 当前节点的左子树指向前一个节点,这一步并不需要判断pre是否为null
cur.left = pre;
// 指针变化
pre = cur;
dfs(cur.right);
}
}
复杂度
- 时间复杂度:O(n),二叉搜索树的中序遍历,每个节点遍历一次
- 空间复杂度:O(n),极端情况下,树退化成为链表,递归深度为n