1. 程式人生 > >leetcode 783. 二叉搜尋樹結點最小距離(遞迴和非遞迴實現java)

leetcode 783. 二叉搜尋樹結點最小距離(遞迴和非遞迴實現java)

題目描述:

給定一個二叉搜尋樹的根結點 root, 返回樹中任意兩節點的差的最小值。

示例:

輸入: root = [4,2,6,1,3,null,null]
輸出: 1
解釋:
注意,root是樹結點物件(TreeNode object),而不是陣列。

給定的樹 [4,2,6,1,3,null,null] 可表示為下圖:

          4
        /   \
      2      6
     / \    
    1   3  

最小的差值是 1, 它是節點1和節點2的差值, 也是節點3和節點2的差值。

注意:

  1. 二叉樹的大小範圍在 2
     到 100
  2. 二叉樹總是有效的,每個節點的值都是整數,且不重複。

思路:中序遍歷得到有序序列並儲存到陣列中,在求相鄰兩元素的最小值

程式碼一、遞迴方法

class Solution {
    public int minDiffInBST(TreeNode root) {
        // if(root==null)
        //     return 0;
        vis=new ArrayList<Integer>();
        dfs(root);
        int min=Integer.MAX_VALUE;
        for(int i=0;i<vis.size()-1;i++){
            min=Math.min(min,vis.get(i+1)-vis.get(i));
        }
        return min;
    }
    List<Integer> vis;
    public void dfs(TreeNode root){
        if(root==null)
            return ;
        else{
            dfs(root.left);
            vis.add(root.val);
            dfs(root.right);
        }
    }
}

程式碼二、非遞迴方法,也很簡單,中序遍歷非遞迴程式碼基本都有模版,只要記住入棧的順序是和前序遍歷結果是一樣的,所以入棧的時候進行操作得到的就是前序遍歷的結果。出棧的順序和中序遍歷的結果是一樣的,所以在出棧的時候進行操作,這些操作得到的就是中序遍歷的結果。剛開始初學前序遍歷的時候無法理解,後來在看了別人的程式碼後自己手寫模擬了幾遍,終於搞懂了中序和前序遍歷,還差後續遍歷還沒搞懂。

class Solution {
   
    public int minDiffInBST(TreeNode root) {
        List<Integer> vis=new ArrayList<Integer>();
        Stack<TreeNode> stack=new Stack<>();

        while(root!=null||!stack.empty()){
            while(root!=null){
                stack.push(root);
                root=root.left;
            }
            if(!stack.empty()){
                TreeNode t=stack.pop();
                vis.add(t.val);
                root=t.right;
            }
        }
        int min=Integer.MAX_VALUE;
        for(int i=0;i<vis.size()-1;i++){
            min=Math.min(min,vis.get(i+1)-vis.get(i));
        }
        return min;
    }
    
}