Morris 前序遍历

1. 前期准备

题目

morris遍历介绍

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;
}