二叉搜索树与双向链表

130 阅读1分钟

题目:leetcode-cn.com/problems/er…

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

由于是二叉搜索树,我们可以直接使用 中序遍历
二叉树的中序遍历

 void dfs(Node node){
        if(node == null) return;
        dfs(node.left);  //左
      	System.out.print(node+","); 中
        dfs(node.right); //右
    }

需要在不创建新节点的情况下去调整树中的节点:
使用 pre 来引用前驱节点,head作为头节点(左子树的最左端的叶子节点,即最小值)

Node pre,head;

使用递归中序遍历树
终止条件: node 为 null,表示没有叶子节点了
在中序的过程 转换为链表

 if(pre == null) head = node;  //遍历到最小值,pre是null,给头节点引用
 else pre.right = node;        //上一个节点的right 引用现节点
 node.left = pre;              //现节点的left 引用 上一个节点 形成双向链表
 pre = node;                   //将pre的引用替换为现节点,作为下一个节点的前驱节点

参考代码:

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

    void dfs(Node node){
        if(node == null) return;
        dfs(node.left);  //左
        if(pre == null) head = node;
        else pre.right = node;     //向右遍历 12345
        node.left = pre;          //向左遍历 54321
        pre = node;   //节点用pre保存
        dfs(node.right); //右
    }
}