剑指 Offer 36. 二叉搜索树与双向链表
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点,只能调整树中节点指针的指向。
为了让您更好地理解问题,以下面的二叉搜索树为例:
4
/ \
2 5
/ \
1 3
我们希望将这个二叉搜索树转化为双向循环链表。链表中的每个节点都有一个前驱和后继指针。对于双向循环链表,第一个节点的前驱是最后一个节点,最后一个节点的后继是第一个节点。
下图展示了上面的二叉搜索树转化成的链表。“head” 表示指向链表中有最小元素的节点。
特别地,我们希望可以就地完成转换操作。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继。还需要返回链表中的第一个节点的指针。
分析
二叉搜索树的中序遍历就是升序序列,所以采用中序遍历递归地进行访问。
题目要求是循环链表,所以我们保存一个头指针head和一个指向当前访问到的节点的指针pre。当访问第一个节点时,给head赋值,并将pre指向它,然后递归地进行访问。
二叉树每个节点都有一个左指针一个右指针,正好符合双向链表的特性。我们在访问到一个节点时,将pre的右指针指向当前节点,并将当前节点的左节点指向pre,就完成了树向链表的转换。
递归结束后,head使用右指针指向了下一个节点,pre是最后一个节点,左指针指向了前驱结点。所以将head的左指针指向pre,pre的右指针指向head,即可形成环形链表。
代码
/*
// 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 {
private Node pre,head;
public Node treeToDoublyList(Node root) {
if(root == null) return null;
recur(root);
//现在pre是最后一个节点,head是头结点
head.left = pre;
pre.right = head;
return head;
}
public void recur(Node root){
if(root == null) return;
recur(root.left);
//现在会一直递归到1这个节点。
//如果有前驱节点
if(pre != null) pre.right = root;
else{
//对于第一个节点,需要先指定头结点
head = root;
}
//将左指针指向前驱结点
//如果是第一个节点,此时pre是null,所以头结点的左指针是空着的。
root.left = pre;
//将前驱结点后移
pre = root;
recur(root.right);
}
}