首页 > 代码库 > LeetCode 99 Recover Binary Search Tree
LeetCode 99 Recover Binary Search Tree
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?
思路;递归。使用递归查找左子树的最大节点,使用递归查找右子树的最小节点;
1若左子树的最大节点的值比右子树的最小节点大,说明是左子树的最大节点与右子树的最小节点进行了交换;
2若若左子树的最大节点的值比根节点大,说明是根节点与左子树的最大节点进行了交换;
3若若右子树的最大节点的值比根节点小,说明是根节点与右子树的最小节点进行了交换;
4若上述三种情况均没有发生,则说明左子树的某两个节点进行了交换或者右子树的某两个节点进行了交换,递归查看左子树和右子树是否有节点进行了交换;
public class Solution { private TreeNode findMin(TreeNode root) { TreeNode result, l = root, r = root; if (root.left != null) l = findMin(root.left); if (root.right!= null) r = findMin(root.right); result = l.val < r.val ? l : r; return result.val < root.val ? result : root; } private TreeNode findMax(TreeNode root) { TreeNode result, l = root, r = root; if (root.left != null) l = findMax(root.left); if (root.right != null) r = findMax(root.right); result = l.val > r.val ? l : r; return result.val > root.val ? result : root; } public void recoverTree(TreeNode root) { TreeNode l=root,r=root; if(root==null||(root.left==null&&root.right==null)) return; if(root.left!=null) l=findMax(root.left); if(root.right!=null) r=findMin(root.right); if(l.val>r.val){ int temp=l.val; l.val=r.val; r.val=temp; return ; }else if(l.val>root.val){ int temp=l.val; l.val=root.val; root.val=temp; return ; }else if(r.val<root.val){ int temp=r.val; r.val=root.val; root.val=temp; return ; }else{ recoverTree(root.left); recoverTree(root.right); } } }
LeetCode 99 Recover Binary Search Tree
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。