给你二叉搜索树的根节点 root ,该树中的两个节点被错误地交换。请在不改变其结构的情况下,恢复这棵树。
进阶:使用 O(n) 空间复杂度的解法很容易实现。你能想出一个只使用常数空间的解决方案吗?
示例 1:
输入:root = [1,3,null,null,2]
输出:[3,1,null,null,2]
解释:3 不能是 1 左孩子,因为 3 > 1 。交换 1 和 3 使二叉搜索树有效。
方法一:中序遍历
对于一棵树,其中序遍历出的值可表示为 [1, 2, 3, 4] 这样的有序数组,如若任意交换两个节点的值,例如 [1, 3, 2, 4],则必有
- 从左子节点截止到最靠近其的第一个被交换节点,遍历得到的值为升序。
- 两被交换节点为倒序,期间可能夹杂升序节点
据此我们可以得出以下代码来查找到两个被交换的节点node1与node2: (其中node1持续更新直到遇到降序节点,node2在node1节点后方查找降序节点)
private long pre = Long.MIN_VALUE;
private TreeNode node1;
private TreeNode node2;
public void recoverTree(TreeNode root) {
search(root);
if (!node1.equals(node2)) {
node1.val ^= node2.val;
node2.val ^= node1.val;
node1.val ^= node2.val;
}
}
private void search(TreeNode root) {
if (root == null) return;
search(root.left);
if (root.val <= pre) {
node2 = root;
} else { // root.val > pre
if (node2 == null) node1 = root;
pre = root.val;
}
search(root.right);
}
方法二:Morris 遍历
能力上来了再更吧orz