算法——二叉搜索树与双向链表

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

解题思路:中序遍历链表,保留上一个节点,每次遍历到一个节点的时候,将上一个节点right指向当前节点,再将当前节点的left指向上个节点。

第一次枚举到left的尽头的时候,就是头节点。

最后需要将头节点和尾节点连上。

class Solution {
    public Node treeToDoublyList(Node root) {
        Deque<Node> stack = new LinkedList<>();
        Node pre = null, head = root;

        if(root == null) return root;

        while(root != null || !stack.isEmpty()) {
            while(root != null) {
                stack.push(root);
                root = root.left;
            }

            root = stack.pop();
            if(pre == null) {
                head = root;
            } else {
                pre.right = root;
                root.left = pre;
            }

            pre = root;

            root = root.right;
        }

        pre.right = head;
        head.left = pre;

        return head;

    }
}
原文地址:https://www.cnblogs.com/lippon/p/14117670.html