想要精通算法和SQL的成长之路 - 恢复二叉搜索树

前言

想要精通算法和SQL的成长之路 - 系列导航

一. 恢复二叉搜索树

原题链接

首先，一个正常地二叉搜索树在中序遍历下，遍历的元素一定是单调递增的。而题目中提醒道，正好有两个节点被互换了。那么以下面的递增序列为例：

• [1,2,3,4,5,6,7]。
• 被替换后：[1,2,6,4,5,3,7]，它一定满足一个特性：存在两个地方，满足递减的情况。
• 第一个问题点：6,4。这里我们需要找到第一个：下一个元素比当前元素小的，我们取当前元素。取前者。
• 第二个问题点：5,3，在找到第一个问题点的前提下，再找：下一个元素比当前元素小的，我们取后者。
• 最后两个问题节点值互换即可（不可以节点互换，一定是值互换）

TreeNode firstNode, secondNode, preNode = new TreeNode(Integer.MIN_VALUE);

public void recoverTree(TreeNode root) {
    inOrderReadTree(root);
    int tmp = firstNode.val;
    firstNode.val = secondNode.val;
    secondNode.val = tmp;
}

public void inOrderReadTree(TreeNode root) {
    if (root == null) {
        return;
    }
    // 左中右的顺序遍历
    inOrderReadTree(root.left);
    // 找到第一个开始递减的元素
    if (firstNode == null && preNode.val > root.val) {
        firstNode = preNode;// 取前者
    }
    // 找到第一个节点的前提下，找到第二个开始递减的元素
    if (firstNode != null && preNode.val > root.val) {
        secondNode = root;// 取后者
    }
    preNode = root;
    inOrderReadTree(root.right);
}