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

31.二叉搜索树与双向链表

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。

要求不能创建任何新的结点，只能调整树中结点指针的指向。

注意：

• 需要返回双向链表最左侧的节点。

例如，输入下图中左边的二叉搜索树，则输出右边的排序双向链表。

思路

二叉搜索树的中序遍历顺序就是一个排好序的序列。想在想的是如何将其连接成双向链表。我的做法是使用一个队列，一边中序遍历一边往队列里存储元素。最后队列中存储着排好序的二叉搜索树的中序遍历序列。

然后就是构造双向链表了，每个出队元素的left指针相当于pre，right指针相当于next。

class Solution {
    Queue<TreeNode> queue=new LinkedList<>();
    public TreeNode convert(TreeNode root) {
        if(root==null) return null;
        dfs(root);
        root=queue.peek();
        TreeNode pre=null;
        TreeNode cur=null;
        while(!queue.isEmpty())
        {
            cur=queue.remove();
            cur.left=pre;
            if(!queue.isEmpty())
            {
                cur.right=queue.peek();
            }else
            {
                cur.right=null;
            }
            pre=cur;
            cur=cur.right;
        }
        return root;
    }
    public void dfs(TreeNode root)
    {
        if(root==null) return;
        dfs(root.left);
        queue.add(root);
        dfs(root.right);
    }
}