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

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

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

思路

  1. 排序列表:二叉搜索树的中序遍历是有序的
  2. 双向链表:设置前驱节点 pre 和当前节点 cur,pre.right = cur, cur.left = pre
  3. 循环链表:head.left = tail, tail.right = head

代码

pre = node 是在访问完左孩子的时候设置的,因为这是中序遍历,我老是想成重建二叉树,这个左右孩子的关系不是原来的二叉树的关系!!!!

class Solution {
    Node pre, head;
    public Node treeToDoublyList(Node root) {
        if (root == null) return null;
        dfs(root);
        head.left = pre;
        pre.right = head;
        return head;
        
    }

    void dfs(Node node) {
        if (node == null) return;
        dfs(node.left);
        
        if (pre != null) pre.right = node;
        else head = node;
        node.left = pre;
        
        pre = node;
        dfs(node.right);
    }
}
posted @ 2022-04-14 16:30  沐灵_hh  阅读(16)  评论(0编辑  收藏  举报