Convert Binary Search Tree to Sorted Doubly Linked List

flyatcmu

已于 2022-04-24 04:43:51 修改

阅读量233

点赞数

分类专栏： Stack 文章标签： leetcode

于 2020-03-02 08:23:09 首次发布

本文链接：https://blog.youkuaiyun.com/u013325815/article/details/104604731

版权

Stack 专栏收录该内容

53 篇文章

订阅专栏

本文介绍了一种将二叉搜索树转换为双向链表的方法，通过中序遍历并利用栈来实现节点之间的连接，最终形成一个循环的双向链表。此方法巧妙地使用了哑节点简化了代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Example 1:

Input: root = [4,2,5,1,3]

Output: [1,2,3,4,5]

思路：Inorder traverse，用stack做；记录一下first 和last node就可以了；最后返回dummpy.right;

错误点：没有想到用dummpy，作为cur node来用；

/*
// 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 {
    public Node treeToDoublyList(Node root) {
        if(root == null) {
            return null;
        }
        Stack<Node> stack = new Stack<>();
        pushLeft(root, stack);
        
        Node first = null;
        Node last = null;
        Node pre = null;
        while(!stack.isEmpty()) {
            Node node = stack.pop();
            // get first;
            if(first == null) {
                first = node;
            }
            // get last;
            if(stack.isEmpty()) {
                last = node;
            }
            // connect middle node;
            if(pre != null) {
                pre.right = node;
                node.left = pre;
            }
            pre = node;
            if(node.right != null) {
                pushLeft(node.right, stack);
            }
        }
        
        first.left = last;
        last.right = first;
        return first;
    }
    
    private void pushLeft(Node node, Stack<Node> stack) {
        while(node != null) {
            stack.push(node);
            node = node.left;
        }
    }
}