リートコードブラッシングの質問-二分探索木のノードの最小距離

Leetcodeでそのような質問を書いた後、私の考えは非常に良いと感じました。それを共有してください。

アイデア：最初に、深さ優先探索によってnull以外の各ノードvalをセットに追加し、次にセットをトラバースして差の最小値を見つけます。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {

    private int min = Integer.MAX_VALUE;

    public int minDiffInBST(TreeNode root) {
        List<Integer> valueList = new ArrayList<Integer>();
        dfs(root,valueList);
        for(int i = 0;i < valueList.size();i++){
            for(int j = i + 1;j < valueList.size();j++){
                min = Math.min(min,Math.abs(valueList.get(i)-valueList.get(j)));
            }
        }
        return min;
    }

    public void dfs(TreeNode node,List<Integer> list){
        if(node != null){
            int a = node.val;
            list.add(a);
            dfs(node.left,list);
            dfs(node.right,list);
        }
    }

}

しかし、それは明らかです。二重のforループは、時間の複雑さを大幅に増加させます。

 

次に、最適化する方法について考える必要があります。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {

    private int min = Integer.MAX_VALUE;

    public int minDiffInBST(TreeNode root) {
        List<Integer> valueList = new ArrayList<Integer>();
        dfs(root,valueList);
        /*for(int i = 0;i < valueList.size();i++){
            for(int j = i + 1;j < valueList.size();j++){
                min = Math.min(min,Math.abs(valueList.get(i)-valueList.get(j)));
            }
        }*/
        Collections.sort(valueList);
        for(int i = 0; i < valueList.size() - 1;i++){
            min = Math.min(min,valueList.get(i+1)-valueList.get(i));
        }
        return min;
    }

    public void dfs(TreeNode node,List<Integer> list){
        if(node != null){
            int a = node.val;
            list.add(a);
            dfs(node.left,list);
            dfs(node.right,list);
        }
    }

}

効果を見てみましょう：