426. Convert Binary Search Tree to Sorted Doubly Linked List
题目地址
题目描述
Convert a Binary Search Tree to a sorted Circular Doubly-Linked List in place.
You can think of the left and right pointers as synonymous to the predecessor and successor pointers in a doubly-linked list. 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.
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. You should return the pointer to the smallest element of the linked list.
Example 1:
Input: root = [4,2,5,1,3]
Output: [1,2,3,4,5]
Explanation: The figure below shows the transformed BST. The solid line indicates the successor relationship, while the dashed line means the predecessor relationship.
Example 2:
Input: root = [2,1,3]
Output: [1,2,3]
Example 3:
Input: root = []
Output: []
Explanation: Input is an empty tree. Output is also an empty Linked List.
Example 4:
Input: root = [1]
Output: [1]
Constraints:
-1000 <= Node.val <= 1000
Node.left.val < Node.val < Node.right.val
All values of Node.val are unique.
0 <= Number of Nodes <= 2000
代码
Approach #1 Recursion
Complexity Analysis
Time complexity : O(N) since each node is processed exactly once.
Space complexity : O(N).
/*
// 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 {
Node first = null;
Node last = null;
public Node treeToDoublyList(Node root) {
if (root == null) return null;
inorder(root);
last.right = first;
first.left = last;
return first;
}
public void inorder(Node node) {
if (node != null) {
inorder(node.left);
if (last != null) {
last.right = node;
node.left = last;
} else {
first = node; // Binary 树最最左边的为 first
}
last = node;
inorder(node.right);
}
}
}
Approach #2
class Solution {
Node prev = null;
public Node treeToDoublyList(Node root) {
if (root == null) return null;
Node dummy = new Node(0, null, null);
prev = dummy;
helper(root);
// connect head and tail
prev.right = dummy.right;
dummy.right.left = prev;
return dummy.right;
}
private void helper (Node cur) {
if (cur == null) return;
helper(cur.left);
prev.right = cur;
cur.left = prev;
prev = cur;
helper(cur.right);
}
}