剑指 offer 36:二叉搜索树与双向链表
代码 01
使用中序遍历访问树的各节点cur,并在访问每个节点的时候构建cur和前驱节点pre的引用指向,中序遍历完成后,最后构建头节点和尾节点引用指向即可。
/*
// 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 head = null;
Node pre = null;
public Node treeToDoublyList(Node root) {
if(root == null){
return null;
}
dfs(root);
head.left = pre;
pre.right = head;
return head;
}
public void dfs(Node root){
if(root == null){
return;
}
dfs(root.left);
if(pre==null){
head = root;
}else{
pre.right = root;
root.left = pre;
}
pre = root;
dfs(root.right);
}
}
时间复杂度
O
(
n
)
O(n)
O(n),
n
n
n是节点个数,需要访问所有的节点
空间复杂度
O
(
n
)
O(n)
O(n),最差情况下,树退化为链表时,递归深度达到
n
n
n,系统使用
O
(
n
)
O(n)
O(n)栈空间。