LeetCode 重构二叉搜索数,即找出两个被交换的节点

来源:互联网 发布:linux中文显示方框 编辑:程序博客网 时间:2024/05/21 09:12

原题: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?

confused what"{1,#,2,3}"means? > read more on how binary tree is serialized on OJ.


OJ's Binary Tree Serialization:

The serialization of a binary tree follows a level order traversal, where '#' signifies a path terminator where no node exists below.

Here's an example:

   1  / \ 2   3    /   4    \     5
The above binary tree is serialized as"{1,2,3,#,#,4,#,#,5}".

题意:二叉搜索树中,有两个结点的位置被交换了,请找出这两个结点并交换回来。

“直观的想法可能是中序遍历一遍二叉树,得到一个有序的二叉树,然后找出其中逆序的地方,交换回来就好了。但这样空间复杂度就是O(n),题目要求O(1)。

我们来分析下有哪些情况:1) 被交换的两个结点相邻,如124356,这样只需要把相邻的3和4交换回来即可;2) 被交换的两个结点不相邻,如163452,这样我们需要找出两个逆序的地方,63和52,并交换第一个逆序的前者和第二个逆序的后者。”(http://blog.csdn.net/ljiabin/article/details/44514651该博主很认真解释了一下,不然光看程序还真看不懂)。因为这两种情况存在,而且这可二叉树只有两个节点被交换了,所以我们需要定位这两个节点,而不是将相邻节点一对一对的交换达到这种效果。这一点也是最精妙的地方,程序中mistake1记录的最底下出错的节点,mistake2记录的是中序遍历中较mistake1后面的节点。


import java.util.*;
/**
 * Definition for binary tree*/
 class TreeNode {
      int val;
      TreeNode left;
      TreeNode right;
      TreeNode(int x) { val = x; 
      }
 }
class Solution {
private TreeNode pre=null;
private TreeNode mistake1,mistake2=null;
public void recoverTree(TreeNode root) {
inOrderTraversal(root);
int temp=mistake2.val;
mistake2.val=mistake1.val;
mistake1.val=temp;
}
/*中序遍历查找出错的两个节点,只有两个节点出错哦
* 左节点值<根节点值<右节点值
* 首先从左子树遍历到底
* */
public void inOrderTraversal(TreeNode root){
if(root==null)
return;
//中序遍历
inOrderTraversal(root.left);//左子树遍历到底
if(pre!=null&&pre.val>root.val){
if(mistake1==null)
mistake1=pre;
mistake2=root;
}
pre=root;//最底下的左节点
inOrderTraversal(root.right);
}

原创粉丝点击