原题链接: http://oj.leetcode.com/problems/recover-binary-search-tree/
这道题是要求恢复一颗有两个元素调换错了的二叉查找树。一開始拿到可能会认为比較复杂,事实上观察出规律了就比較简单。主要还是利用二叉查找树的主要性质,就是中序遍历是有序的性质。那么假设当中有元素被调换了,意味着中序遍历中必定出现违背有序的情况。那么会出现几次呢?有两种情况,假设是中序遍历相邻的两个元素被调换了,非常easy想到就仅仅需会出现一次违反情况,仅仅须要把这个两个节点记录下来最后调换值就能够;假设是不相邻的两个元素被调换了,举个样例非常easy能够发现,会发生两次逆序的情况,那么这时候须要调换的元素应该是第一次逆序前面的元素,和第二次逆序后面的元素。比方1234567,1和5调换了,会得到5234167,逆序发生在52和41,我们须要把4和1调过来,那么就是52的第一个元素,41的第二个元素调换就可以。
搞清楚了规律就easy实现了,中序遍历寻找逆序情况,调换的第一个元素,永远是第一个逆序的第一个元素,而调换的第二个元素假设仅仅有一次逆序,则是那一次的后一个,假设有两次逆序则是第二次的后一个。算法仅仅须要一次中序遍历,所以时间复杂度是O(n),空间是栈大小O(logn)。代码例如以下:
这道题还是考察二叉树遍历,只是应用题目要求套了一个不同的外壳,须要我们利用二叉查找树的性质观察出规律之后才干求解。
这道题是要求恢复一颗有两个元素调换错了的二叉查找树。一開始拿到可能会认为比較复杂,事实上观察出规律了就比較简单。主要还是利用二叉查找树的主要性质,就是中序遍历是有序的性质。那么假设当中有元素被调换了,意味着中序遍历中必定出现违背有序的情况。那么会出现几次呢?有两种情况,假设是中序遍历相邻的两个元素被调换了,非常easy想到就仅仅需会出现一次违反情况,仅仅须要把这个两个节点记录下来最后调换值就能够;假设是不相邻的两个元素被调换了,举个样例非常easy能够发现,会发生两次逆序的情况,那么这时候须要调换的元素应该是第一次逆序前面的元素,和第二次逆序后面的元素。比方1234567,1和5调换了,会得到5234167,逆序发生在52和41,我们须要把4和1调过来,那么就是52的第一个元素,41的第二个元素调换就可以。
搞清楚了规律就easy实现了,中序遍历寻找逆序情况,调换的第一个元素,永远是第一个逆序的第一个元素,而调换的第二个元素假设仅仅有一次逆序,则是那一次的后一个,假设有两次逆序则是第二次的后一个。算法仅仅须要一次中序遍历,所以时间复杂度是O(n),空间是栈大小O(logn)。代码例如以下:
public void recoverTree(TreeNode root) { if(root == null) return; ArrayList<TreeNode> pre = new ArrayList<TreeNode>(); pre.add(null); ArrayList<TreeNode> res = new ArrayList<TreeNode>(); helper(root,pre, res); if(res.size()>0) { int temp = res.get(0).val; res.get(0).val = res.get(1).val; res.get(1).val = temp; } } private void helper(TreeNode root, ArrayList<TreeNode> pre, ArrayList<TreeNode> res) { if(root == null) { return; } helper(root.left, pre, res); if(pre.get(0)!=null && pre.get(0).val>root.val) { if(res.size()==0) { res.add(pre.get(0)); res.add(root); } else { res.set(1,root); } } pre.set(0,root); helper(root.right,pre,res); }能够看到实现中pre用了一个ArrayList来存,这样做的原因是由于java都是值传递,所以我们要全局变化pre的值(而不是在当前函数里),仅仅能传一个数组,才干改变结点的地址,这一点很重要,也是java和C++一个比較大的差别,不了解的朋友能够研究一下哈。
这道题还是考察二叉树遍历,只是应用题目要求套了一个不同的外壳,须要我们利用二叉查找树的性质观察出规律之后才干求解。