剑指 Offer 36. 二叉搜索树与双向链表（DFS&&头指针）

剑指 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);
    }
}