恢复二叉搜索树
题目描述:
给你二叉搜索树的根节点 root ,该树中的两个节点被错误地交换。请在不改变其结构的情况下,恢复这棵树。进阶:使用 O(n) 空间复杂度的解法很容易实现。你能想出一个只使用常数空间的解决方案吗?提示:树上节点的数目在范围 [2, 1000] 内-2 ^31 <= Node.val <= 2 ^31 - 1
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public void recoverTree(TreeNode root) {
List<Integer> nums = new ArrayList<Integer>();
inorder(root, nums);
int[] swapped = findTwoSwapped(nums);
recover(root, 2, swapped[0], swapped[1]);
}
// 中序遍历
public void inorder(TreeNode root, List<Integer> nums) {
if (root == null) {
return;
}
inorder(root.left, nums);
nums.add(root.val);
inorder(root.right, nums);
}
// 确定交换的位置
public int[] findTwoSwapped(List<Integer> nums) {
int n = nums.size();
int x = -1, y = -1;
for (int i = 0; i < n - 1; ++i) {
if (nums.get(i + 1) < nums.get(i)) {
y = nums.get(i + 1);
if (x == -1) {
x = nums.get(i);
} else {
break;
}
}
}
return new int[]{x, y};
}
// 恢复元素的位置
public void recover(TreeNode root, int count, int x, int y) {
if (root != null) {
if (root.val == x || root.val == y) {
root.val = root.val == x ? y : x; //交换元素
if (--count == 0) { // 结束条件
return;
}
}
recover(root.right, count, x, y);
recover(root.left, count, x, y);
}
}
}
按照官方的题解,要明确二叉搜索树的定义,我们知道二叉树中序遍历的结果一定是按照顺序排列的,故只要其中两个节点发生值的交换,就可以说明该二叉搜索树中的两个节点发生了改变。利用该特性来解决该问题。详细请看代码,有疑问欢迎留言。