二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树。
示例 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
思路:我们可以考虑到上一题中的方法三:验证二叉搜索树,采用中序遍历的方法。因为我们知道,一个合格的二叉搜索树在中序遍历后得到的序列一定是一个升序序列,当有两个点被交换时一定会出现两次前一个点或者一次(相邻节点被交换了)前一个点比后一个点大的情况,因此我们找到这两个点,然后保存一下,之后交换回来即可, 因为只用交换值,所以这题中规中矩吧。
class Solution {
public void recoverTree(TreeNode root) {
Stack<TreeNode> st=new Stack<TreeNode>();
TreeNode firstNode=null,secondNode=null;
TreeNode pre=new TreeNode(Integer.MIN_VALUE);
while(!st.isEmpty() || root!=null)
{
while(root!=null)
{
st.push(root);
root=root.left;
}
root=st.pop();
if(firstNode==null && pre.val>=root.val) firstNode=pre;
if(firstNode!=null && pre.val>=root.val) secondNode=root;
pre=root;
root=root.right;
}
int tmp=firstNode.val;
firstNode.val=secondNode.val;
secondNode.val=tmp;
}
}