算法學習之二叉搜索樹與雙向鏈表

題目描述

輸入一棵二叉搜索樹，將該二叉搜索樹轉換成一個排序的雙向鏈表。要求不能創建任何新的結點，只能調整樹中結點指針的指向。

思路很簡單，但剛度題目時有些懵逼：中序遍歷二叉樹，然後調整指針即可。

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
import java.util.*;
public class Solution {
    public TreeNode Convert(TreeNode pRootOfTree) {
           if(pRootOfTree==null)return null;
           ArrayList<TreeNode> list = new ArrayList<TreeNode>();
           getList(pRootOfTree,list);
           return getPoint(list);
    }
    public void getList(TreeNode root,ArrayList<TreeNode> list){
        if(root.left!=null){
            getList(root.left,list);
        }
        list.add(root);
        if(root.right!=null){
            getList(root.right,list);
        }
        
    }
    public TreeNode getPoint(ArrayList<TreeNode> list){
        for(int i =0 ;i<list.size()-1;i++){
            list.get(i).right = list.get(i+1);
            list.get(i+1).left = list.get(i);
        }
        return list.get(0);
    }
}

• 時間複雜度：O(n)。
• 空間複雜度：O(logn)。