zoukankan      html  css  js  c++  java
  • [LeetCode#99]Recover Binary Search Tree

    The problem:

    Two elements of a binary search tree (BST) are swapped by mistake.

    Recover the tree without changing its structure.

    Note:
    A solution using O(n) space is pretty straight forward. Could you devise a constant space solution?

    My analysis:

    The idea behind this problem is so elegant!!!
    The key idea: we can get the swapped nodes from the inroder traversal series of the binary tree. It includes following two conditions:
    1. the two swapped nodes in neighbour with each other.
    Inorder traversal series: 1 ,2 ,3 ,5 ,4, 6, 7

    2. the two swapped nodes are not in neighbour with each other.
    Inorder traversal series: 1, 5, 3, 4, 2, 6, 7

    The two cases needed to be tackled differently.
    In situation 1, we would detect only one violation, the swapped nodes should be 5, 4
    In situation 2, we would detect two violations, the swapped nodes should be: the pre node(5) in the first violation, and the post node(2) in the second violation.

    skills in implementation:
    1. Since Java pass arguments by value, we should use the ArrayList to record the target nodes we have figured out during the recursion process.
    2. Swap nodes by value rather than by pointer. (don not fool in thinking about recording the parent's pointer)

    My solution:

    public class Solution {
        public void recoverTree(TreeNode root) {
            
            if (root == null) //note. the declaration's return value is void 
                return;
            
            ArrayList<TreeNode> pre = new ArrayList<TreeNode> ();
            ArrayList<TreeNode> ret = new ArrayList<TreeNode> ();
            pre.add(null);
            
            helper(root, pre, ret);//inorder to track the result, we use ArrayList.
            
            if (ret.size() > 0) {
                
                int temp = ret.get(0).val;
                ret.get(0).val = ret.get(1).val;
                ret.get(1).val = temp;
            }
            
            return;
        }
        
        private void helper(TreeNode root, ArrayList<TreeNode> pre, ArrayList<TreeNode> ret) {
            
            if (root == null)
                return;
            
            helper(root.left, pre, ret);
            
            if (pre.get(0) != null && pre.get(0).val >= root.val) {
                if (ret.size() == 0) {
                    ret.add(pre.get(0));
                    ret.add(root);
                } else {//we have already detected a violation 
                    ret.set(1, root);
                }
            }
            pre.set(0, root);
            
            helper(root.right, pre, ret);
            
            return;
        }
    }
  • 相关阅读:
    Python基础07
    python基础06
    python基础05
    python基础04
    python基础03
    python基础02
    python组件之wtforms
    PythonWeb框架之Flask
    Linux下yum安装Redis
    在vuex的mutations中使用vue的小技巧
  • 原文地址:https://www.cnblogs.com/airwindow/p/4212332.html
Copyright © 2011-2022 走看看