1. 前期准备
2. 题目描述
二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树。
示例 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(1) 空间复杂度
3. 解析
下面以一个二叉树的morris遍历为例
可以看出使用morris遍历得到的节点顺序如上图所示为1 2 3 2 4 1 5 6 5 1
,这棵二叉树的中序遍历的顺序为 3 2 4 1 6 5 1
,对照上面的序列 很容易看出,第二次出现的节点的位置的顺序就是中序遍历。由此结合morris遍历规则,可以知道在指针右移时,记录节点值,就是树的中序遍历!
对于二叉搜索树来说,中序遍历是从小到大依次上升的顺序,如果改变其中两个值,必定会有两个逆序对即 pre > cur
,这里需要进行讨论:
- 第一次出现:记录pre
- 第二次出现:记录cur
恢复的过程就是将两个记录节点中值的顺序进行互换。
4. 代码
public void recoverTree(TreeNode root) {
if (root == null) {
return;
}
TreeNode cur = root;
TreeNode[] t = new TreeNode[2];
TreeNode pre = null;
while (cur != null) {
if (cur.left != null) {
TreeNode mostright = cur.left;
while (mostright.right != null && mostright.right != cur) {
mostright = mostright.right;
}
if (mostright.right == null) {
mostright.right = cur;
cur = cur.left;
} else {
// 指针右移的位置
if (pre!=null && pre.val > cur.val) {
if (t[0] == null){
t[0] = pre;
}
t[1] = cur;
}
mostright.right = null;
pre = cur;
cur = cur.right;
}
} else {
// 指针右移的位置
if (pre!=null && pre.val > cur.val) {
if (t[0] == null){
t[0] = pre;
}
t[1] = cur;
}
pre = cur;
cur = cur.right;
}
}
if (t[0] == null || t[1] == null) {
return;
}
// 交换节点
int tmp = t[0].val;
t[0].val = t[1].val;
t[1].val = tmp;
// System.out.println(t[0].val + " " + t[1].val);
return;
}