delete an extra edge from BST

Given a BST, remove an extra edge to make it BST.

    7

   / \

  5   9

 /  \ /  

3    8

 1 public void deleteEdge(TreeNode root) {
 2     if(root == null) return;
 3     root = dfs(root, Integer.MIN_VALUE, Integer.MAX_VALUE);
 4 }
 5 private TreeNode dfs(TreeNode root, int left, int right) {
 6     if(root == null) return null;
 7     if(root.val <= left || root.val >= right) return null;
 8     root.left = dfs(root.left, left, root.val);
 9     root.right = dfs(root.right, root.val, right);
10     return root;
11 }

猜你喜欢

转载自www.cnblogs.com/beiyeqingteng/p/11068227.html
BST