力扣 剑指 Offer 36. 二叉搜索树与双向链表 打卡

剑指 Offer 36. 二叉搜索树与双向链表

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点，只能调整树中节点指针的指向。

为了让您更好地理解问题，以下面的二叉搜索树为例：

我们希望将这个二叉搜索树转化为双向循环链表。链表中的每个节点都有一个前驱和后继指针。对于双向循环链表，第一个节点的前驱是最后一个节点，最后一个节点的后继是第一个节点。

下图展示了上面的二叉搜索树转化成的链表。“head” 表示指向链表中有最小元素的节点。

特别地，我们希望可以就地完成转换操作。当转化完成以后，树中节点的左指针需要指向前驱，树中节点的右指针需要指向后继。还需要返回链表中的第一个节点的指针。

解题思路：同上一篇文章转换成单链表，只不过那个是先序遍历，这个是中序遍历，中序遍历二叉搜索树的结果就是有序的，然后再去构造双向链表，运用三个标志结点，returnNode，pre，index。来完成整个双向链表的构造，最后返回returnNode的下一个结点即可。代码和提交结果如下图： 

 

class Solution {
    public Node treeToDoublyList(Node root) {
        if(root == null){
            return null;
        }
        List<Node> list = new ArrayList();
        //中序遍历得到二叉搜索树的结点，按顺序存在List当中
            dfs(root,list);
            int len = list.size();
            //创建一个虚拟头节点returnNode，然后其余的两个结点 pre 和 index 是用来进行双向链表
            //的构造用的
            Node returnNode = new Node(-1);
            Node pre = returnNode;
            Node index = returnNode;
            for(int i = 0 ; i < len ; i++){
                //首先先让pre指向下一个结点
                pre.right = list.get(i);
                //然后把pre移动到下一个结点
                pre = pre.right;
                //然后让他的左指针指向index，
                pre.left = index;
                //然后让index到下一个结点，也就是pre的位置，重复完List内所有的结点
                index = index.right;
            }
            //最后就是把List内最后一个结点的右指针指向第一个结点，
            //然后第一个结点的左指针指向最后一个结点，和循环里面一样的套路
            pre.right = returnNode.right;
            pre = pre.right;
            pre.left = index;
            //我们的虚拟头节点returnNode的下一个结点就是我们要返回的结点
            return returnNode.right;    
    } 
    private void dfs(Node root,List list){
        if(root == null){
            return;
        }
        dfs(root.left,list);
        list.add(root);
        dfs(root.right,list);
        return;
    }
}

提交结果：