LeetCode-99. 恢复二叉搜索树-Java-medium

题目链接

法一（中序遍历）

public class Solution99 {

    private TreeNode first;  // 中序遍历中第一次遇到的被错误交换的结点（较大的）
    private TreeNode second; // 中序遍历中第二次遇到的被错误交换的结点（较小的）
    private TreeNode pre;    // 记录中序遍历过程中当前结点的前驱

    /**
     * 无参构造函数
     */
    public Solution99() {
        this.first = null;
        this.second = null;
        this.pre = null;
    }

    /**
     * 在中序遍历的过程中记录被错误交换的结点
     *
     * @param root
     */
    private void inorderTraversal(TreeNode root) {
        if (root == null) {
            return;
        }
        inorderTraversal(root.left);
        if (pre != null && pre.val > root.val) { // 不满足前一个元素 < 当前元素
            if (first == null) {
                first = pre; // first是被错误交换的结点中较大的
            }
            second = root; // second是被错误交换的结点中较小的
        }
        pre = root; // pre按中序遍历的次序向后移动
        inorderTraversal(root.right);
    }

    /**
     * 交换Tree两个结点的值
     */
    private void swap() {
        if (first != null && second != null && first.val != second.val) { // 两个相同的数^结果为0
            first.val ^= second.val;
            second.val ^= first.val;
            first.val ^= second.val;
        }
    }

    /**
     * 法一（中序遍历）
     * 空间复杂度：O(logn)
     *
     * @param root
     */
    public void recoverTree(TreeNode root) {
        inorderTraversal(root);
        swap();
    }
}

本地测试

        /**
         * 99. 恢复二叉搜索树
         */
        lay.showTitle(99);
        Solution99 sol99 = new Solution99();
        List<Integer> arr99 = Arrays.asList(3, 1, 4, null, null, 2);
        TreeNode root99 = treeOpt.createTreeByLayerOrder(arr99);
        treeOpt.layerOrder(root99);
        sol99.recoverTree(root99);
        treeOpt.layerOrder(root99);