题目描述:
二叉搜索树(BST)中的两个节点被错误地交换了, 请在不改变树的结构的情况下恢复这棵树。
备注: 用O(n)的空间解决这个问题的方法太暴力了,你能设计一个常数级空间复杂度的算法么?
解题思路:
- 这道题我想的有点复杂,我是想着怎么交换这两个节点
- 实际上怎么做呢?只需要找到这两个出错的节点,交换它们的值就好
代码如下:
public void recoverTree(TreeNode root) {
//可以和上一题 的思路一样,还是通过中序遍历 来做
ArrayList<TreeNode> list = new ArrayList<TreeNode>();
if(root == null){
return;
}
inorder(root,list);
//找到两个出错的节点
int i;
int j;
for ( i = 0; i < list.size() - 1; i ++) {
if(list.get(i).val > list.get(i + 1).val) {
break;
}
}
for (j = list.size() - 1; j > 0; j --) {
if(list.get(j).val < list.get(j - 1).val) {
break;
}
}
// 交换两个出错节点的值就好
int temp = list.get(i).val;
list.get(i).val = list.get(j).val;
list.get(j).val = temp;
}
// 中序遍历得到该二叉树的有序数组
private static void inorder(TreeNode root,List list){
if(root != null){
inorder(root.left,list);
list.add(root);
inorder(root.right,list);
}
}