426. Convert Binary Search Tree to Sorted Doubly Linked List

426. Convert Binary Search Tree to Sorted Doubly Linked List


Convert a BST to a sorted circular doubly-linked list in-place. Think of the left and right pointers as synonymous to the previous and next pointers in a doubly-linked list.

Let’s take the following BST as an example, it may help you understand the problem better:

426. Convert Binary Search Tree to Sorted Doubly Linked List

We want to transform this BST into a circular doubly linked list. Each node in a doubly linked list has a predecessor and successor. For a circular doubly linked list, the predecessor of the first element is the last element, and the successor of the last element is the first element.

The figure below shows the circular doubly linked list for the BST above. The “head” symbol means the node it points to is the smallest element of the linked list.

426. Convert Binary Search Tree to Sorted Doubly Linked List

Specifically, we want to do the transformation in place. After the transformation, the left pointer of the tree node should point to its predecessor, and the right pointer should point to its successor. We should return the pointer to the first element of the linked list.

The figure below shows the transformed BST. The solid line indicates the successor relationship, while the dashed line means the predecessor relationship.

426. Convert Binary Search Tree to Sorted Doubly Linked List

方法1: inorder traversal

思路:

使用inorder traversal,每次pop出节点我们要做这样几步:把prev -> right 指向top,top ->left 指向prev。最后一步,把最后一个top节点以此方法和head进行双向连接。

易错点

  1. never, never leaves ptrs unitialized
/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* left;
    Node* right;

    Node() {}

    Node(int _val, Node* _left, Node* _right) {
        val = _val;
        left = _left;
        right = _right;
    }
};
*/
class Solution {
public:
    Node* treeToDoublyList(Node* root) {
        if (!root) return nullptr;
        Node * cur = root, * head = nullptr, * prev = nullptr;
        stack<Node*> st;
        
        while (true) {
            if (root) {
                st.push(root);
                root = root -> left;
            }
            else {
                if (st.empty()) break;
                
                root = st.top();
                st.pop();
                
                if (!head) head = root;
                if (prev) {
                    root -> left = prev;
                    prev -> right = root;
                }
                prev = root;
                root = root -> right;
            }
        }
        head -> left = prev;
        prev -> right = head;
        return head;
    }
};