99. 恢复二叉搜索树
二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树。
示例 1:
输入: [1,3,null,null,2]
1
/
3
\
2
输出: [3,1,null,null,2]
3
/
1
\
2
示例 2:
输入: [3,1,4,null,null,2]
3
/ \
1 4
/
2
输出: [2,1,4,null,null,3]
2
/ \
1 4
/
3
进阶:
- 使用 O(n) 空间复杂度的解法很容易实现。
- 你能想出一个只使用常数空间的解决方案吗?
不好意思我想不出常数空间的,只能去看题解了。自己写的是O(N)的。感觉也还可以吧
class Solution {
List<Integer> list = new ArrayList<>();
public void inorder(TreeNode root){
if(root == null)
return;
inorder(root.left);
list.add(root.val);
inorder(root.right);
}
public void recover(TreeNode root,int x ,int y){
if(root == null)
return;
if(root.val == x || root.val == y){
if(root.val == x)
root.val = y;
else
root.val = x;
}
recover(root.left, x , y);
recover(root.right, x , y);
}
public void recoverTree(TreeNode root) {
inorder(root);
int x = 0, y = 0;
int len = list.size();
for(int i = 0 ; i < len-1 ; i++)
if(list.get(i) > list.get(i+1)){
x = list.get(i);
break;
}
for(int i = len-1 ; i > 0 ; i--)
if(list.get(i) < list.get(i-1)){
y = list.get(i);
break;
}
recover(root,x,y);
}
}