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