首页 > 代码库 > [Leetcode][JAVA] Recover Binary Search Tree (Morris Inorder Traversal)

[Leetcode][JAVA] Recover Binary Search Tree (Morris Inorder Traversal)

Two elements of a binary search tree (BST) are swapped by mistake.

Recover the tree without changing its structure.

Note:
A solution using O(n) space is pretty straight forward. Could you devise a constant space solution?

 

使用O(n)空间的话可以直接中序遍历来找问题节点。

如果是O(1)空间的话,基本上就只能是原地操作了。

这里介绍一个Morris Inorder Traversal。可以实现:

1. 如果当前节点有左子树,那么找到左子树的最右节点并把它右指针指向当前节点。当前节点转移到其左节点。

2. 如果左子树的最右节点已经指向了当前节点(之前已经遍历过左子树),那么还原最右节点的右指针为null, 输出当前节点,当前节点转移到其右节点。

3. 如果当前节点没有左子树,那么直接输出当前节点并将其转移到右节点。

写成代码如下:

 1     public void recoverTree(TreeNode root) { 2         TreeNode cur = root; 3         while(cur!=null) { 4             if(cur.left!=null) { 5                 TreeNode temp = cur.left; 6                 while(temp.right!=null && temp.right!=cur) 7                     temp = temp.right; 8                 if(temp.right==null) { 9                     temp.right = cur;10                     cur = cur.left;11                 }12                 else {13                     temp.right=null;14                     //print15                     cur=cur.right;16                 }17             }18             else {19                 //print20                 cur=cur.right21             }22         }23     }24 }

 

结合这道题,只需要在print的地方添加代码即可。

只存在一处错误,遍历时可能出现两种情况:

1. 发现一次原先节点值比当前节点值大,例如:(1, 4, 3, 7, 9)这时只有对调这两个节点值即可。

2. 发现两次原先节点值比当前节点值大,例如: (1, 9, 4, 5, 3, 10)这时需要对调第一次的原先节点值和第二次的当前节点值。

使用两个TreeNode wrong1, wrong2记录这两个错误节点,如果是第一次发现原先节点比当前节点值大的错误,则wrong1置为原先节点,wrong2置为当前节点。如果又发现一次,则只更改wrong2.

注意原先节点pre和当前节点cur的关系,只有cur即将挪向右节点之前才将pre置为cur. (因为cur挪向左节点是第一次遍历左子树,仅仅用来连接,第二次遍历才输出)

完整代码如下:

 1     public void recoverTree(TreeNode root) { 2         TreeNode cur = root; 3         TreeNode pre = null; 4         TreeNode wrong1 = null; 5         TreeNode wrong2 = null; 6         while(cur!=null) { 7             if(cur.left!=null) { 8                 TreeNode temp = cur.left; 9                 while(temp.right!=null && temp.right!=cur)10                     temp = temp.right;11                 if(temp.right==null) {12                     temp.right = cur;13                     cur = cur.left;14                 }15                 else {16                     temp.right=null;17                     //print18                     if(pre!=null && pre.val>cur.val) {19                         if(wrong1==null)20                             wrong1=pre;21                         wrong2 = cur;22                     }23                     pre = cur;24                     cur=cur.right;25                 }26             }27             else {28                 //print29                 if(pre!=null && pre.val>cur.val) {30                     if(wrong1==null)31                         wrong1=pre;32                     wrong2 = cur;33                 }34                 pre = cur;35                 cur=cur.right;36             }37         }38         int t = wrong1.val;39         wrong1.val = wrong2.val;40         wrong2.val = t;41     }

 

[Leetcode][JAVA] Recover Binary Search Tree (Morris Inorder Traversal)