剑指 Offer 36. 二叉搜索树与双向链表

题目

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点，只能调整树中节点指针的指向。

为了让您更好地理解问题，以下面的二叉搜索树为例：

我们希望将这个二叉搜索树转化为双向循环链表。链表中的每个节点都有一个前驱和后继指针。对于双向循环链表，第一个节点的前驱是最后一个节点，最后一个节点的后继是第一个节点。

下图展示了上面的二叉搜索树转化成的链表。“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