剑指 Offer 36. 二叉搜索树与双向链表
剑指 Offer 36. 二叉搜索树与双向链表
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点,只能调整树中节点指针的指向。
思路
- 排序列表:二叉搜索树的中序遍历是有序的
- 双向链表:设置前驱节点 pre 和当前节点 cur,pre.right = cur, cur.left = pre
- 循环链表:head.left = tail, tail.right = head
代码
pre = node 是在访问完左孩子的时候设置的,因为这是中序遍历,我老是想成重建二叉树,这个左右孩子的关系不是原来的二叉树的关系!!!!
class Solution {
Node pre, head;
public Node treeToDoublyList(Node root) {
if (root == null) return null;
dfs(root);
head.left = pre;
pre.right = head;
return head;
}
void dfs(Node node) {
if (node == null) return;
dfs(node.left);
if (pre != null) pre.right = node;
else head = node;
node.left = pre;
pre = node;
dfs(node.right);
}
}