leetcode 783. 二叉搜尋樹結點最小距離(遞迴和非遞迴實現java)
阿新 • • 發佈:2018-11-09
題目描述:
給定一個二叉搜尋樹的根結點 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的差值。
注意:
- 二叉樹的大小範圍在
2
100
。 - 二叉樹總是有效的,每個節點的值都是整數,且不重複。
思路:中序遍歷得到有序序列並儲存到陣列中,在求相鄰兩元素的最小值
程式碼一、遞迴方法
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; } }