一、问题

一个二叉搜索树的根节点 root ,该树中的 恰好 两个节点的值被错误地交换。请在不改变其结构的情况下,恢复这棵树 。

示例 1:

恢复二叉搜索树_数组

输入:root = [1,3,null,null,2]
输出:[3,1,null,null,2]
解释:3 不能是 1 的左孩子,因为 3 > 1 。交换 1 和 3 使二叉搜索树有效。

示例 2:

恢复二叉搜索树_二叉搜索树_02

输入:root = [3,1,4,null,null,2]
输出:[2,1,4,null,null,3]
解释:2 不能在 3 的右子树中,因为 2 < 3 。交换 2 和 3 使二叉搜索树有效。

 二、题解

解法一

注意题目给出的条件,是 二叉搜索树,这就是意味着节点之间是有顺序关系的。

如果我们把整棵树都 遍历 一遍,将遍历的结果保存下来,比如放到一个数组中。

那么这个数组应该是有序的。

既然是有序的那就好办了,我们将这个有序的数组遍历一遍。

如果数组是完全有序的,那么直接返回就可以了。

否则,我们找到顺序不一致的两个下标 i 和 j,将 arr[i].val 和 arr[j].val 的值互换一下即可。

代码实现:

class Solution {
    public void recoverTree(TreeNode root) {
        List<TreeNode> list = new ArrayList<TreeNode>();
        dfs(root,list);
        TreeNode x = null;
        TreeNode y = null;
        //扫面遍历的结果,找出可能存在错误交换的节点x和y
        for(int i=0;i<list.size()-1;++i) {
            if(list.get(i).val>list.get(i+1).val) {
                y = list.get(i+1);
                if(x==null) {
                    x = list.get(i);
                }
            }
        }
        //如果x和y不为空,则交换这两个节点值,恢复二叉搜索树
        if(x!=null && y!=null) {
            int tmp = x.val;
            x.val = y.val;
            y.val = tmp;
        }
    }

    //中序遍历二叉树,并将遍历的结果保存到list中        
    private void dfs(TreeNode node,List<TreeNode> list) {
        if(node==null) {
            return;
        }
        dfs(node.left,list);
        list.add(node);
        dfs(node.right,list);
    }
}

解法二

解法一种,我们利用了额外的数组保存了遍历的结果,如果后面一个数比前面一个数小,那就找到了要交换的节点。

按照同样的思路,用中序遍历的方式遍历这颗二叉搜索树,我们再增加一个辅助的 pre 指针,记录 上一个 节点的值。

如果 当前节点的值,小于 上一个节点的值,这就找到了需要交换的节点。利用这种方式,就不需要额外的数组空间了。

注意,这种方式仍然使用了外部空间,虽然我们只用了常数个变量,但是递归调用仍然是需要额外空间的,其空间复杂度是 O(h)O(h)O(h),h 为树的高度。

所以用递归的方式遍历,或者手动模拟栈的方式遍历,都没有达到真正的常数空间。

代码实现 :

class Solution {
    //用两个变量x,y来记录需要交换的节点
    private TreeNode x = null;
    private TreeNode y = null;
    private TreeNode pre = null;
    public void recoverTree(TreeNode root) {
        dfs(root);
        //如果x和y都不为空,说明二叉搜索树出现错误的节点,将其交换
        if(x!=null && y!=null) {
            int tmp = x.val;
            x.val = y.val;
            y.val = tmp;
        }
    }
	
    //中序遍历二叉树,并比较上一个节点(pre)和当前节点的值,如果pre的值大于当前节点值,则记录下这两个节点
    private void dfs(TreeNode node) {
        if(node==null) {
            return;
        }
        dfs(node.left);
        if(pre==null) {
            pre = node;
        }
        else {
            if(pre.val>node.val) {
                y = node;
                if(x==null) {
                    x = pre;
                }
            }
            pre = node;
        }
        dfs(node.right);
    }
}

解法三

解法二还不是真正的常数空间复杂度,想要达到常数空间,我们可以用 莫里斯遍历,这种方式可以做到 O(1)O(1)O(1) 的空间复杂度去遍历一棵树。

我们先看看莫里斯遍历到底是咋回事

回想一下中序遍历的递归版本是

dfs(root.left)

打印节点 root

dfs(root.right)

也就是一路往左走到底,左边走不通了,再往右边走。对于上图来说,就是 4 -> 3 -> 1 这个过程,一路往左,走不通了再往右,也就是遍历 2。

当然如果 2 的右边还有节点那么还会继续遍历下去。

现在 2 的右边已经是空了,对于递归来说操作系统自动出栈,然后会访问 3 这个节点。

既然 2 是叶子节点,左右子树都是空,我们可以利用这个空闲出来的信息,将 2 的右子树指向 3,这样当 2 遍历完后,再往右走,就会自动走到 3 这个节点了。

同理,将 3 的右子树指向 4,将 6 的右子树指向 7。

这样的话,我们就可以省去额外的栈空间了。利用叶子节点的右子树这个特点,将其重新赋予指向关系 ,就是莫里斯遍历的核心了。

不过光是这样还不行,再回看上面那张图,其实已经不是一棵树了,而是变成图了。因为出现了循环。

所以,我们还需要将新加这个指向关系给去掉。

对于上图来说,假设我们已经遍历到 4 这个节点了,那就意味着 4 在左子树都遍历完了,对应的就是 1,2,3 都遍历完了。

3.right=4 这个是我们新加上的,既然现在已经遍历到 4 了,我们就可以将 3.right=null,将这个指向关系还原即可。

从上图中也可以看出,所谓新加的指向关系,就是找到根节点左子树的最右子树,然后将最右子树的 right 指向根节点。

class Solution {
	public void recoverTree(TreeNode root) {
		if(root==null) {
			return;
		}
		TreeNode x = null;
		TreeNode y = null;
		TreeNode pre = null;
		TreeNode tmp = null;
		while(root!=null) {
			if(root.left!=null) {
				tmp = root.left;
				while(tmp.right!=null && tmp.right!=root) {
					tmp = tmp.right;
				}
				if(tmp.right==null) {
					tmp.right = root;
					root = root.left;
				}
				else {
					if(pre!=null && pre.val>root.val) {
						y = root;
						if(x==null) {
							x = pre;
						}
					}
					pre = root;
					tmp.right = null;
					root = root.right;
				}
			}
			else {
				if(pre!=null && pre.val>root.val) {
					y = root;
					if(x==null) {
						x = pre;
					}
				}
				pre = root;
				root = root.right;
			}
		}
		if(x!=null && y!=null) {
			int t = x.val;
			x.val = y.val;
			y.val = t;
		}
	}
}