题目要求:
Two elements of a binary search tree (BST) are swapped by mistake.
Recover the tree without changing its structure.
给出一棵二叉树,其中有两个结点交换了位置,要求找出这两个结点,恢复正确的二叉树。
注意:要求空间复杂度为O(1)
思路:
由于二叉树的中序遍历是升序排列的,因此当交换了两个结点后,再次遍历该二叉树会出现乱序现象。
第一次出现乱序的地方,就是要找的第一个结点;
最后一次出现乱序的地方,就是要找的第二个结点(注意此处说的是最后一次乱序的地方,而不是第二次乱序的地方)。
代码:
if(root == null){
return;
}
//利用中序遍历的上升特性
Stack<TreeNode> ts = new Stack<TreeNode>();
TreeNode ptr = root;
//记录第一次出现乱序的地方
boolean flag = true;
TreeNode sw1 = null, sw2 = null, pre = null;
while( ptr != null || !ts.isEmpty()){
while(ptr != null){
ts.push(ptr);
ptr = ptr.left;
}
if(!ts.isEmpty()){
//栈顶弹栈
ptr = ts.pop();
//第一次出现乱序的地方
if(pre!= null && flag && pre.val > ptr.val){
//此时错误结点为pre
sw1 = pre;
flag = false;
}
//最后一次出现乱序的地方
if(pre != null && pre.val > ptr.val){
//此时错误结点为当前结点ptr
sw2 = ptr;
}
pre = ptr;
ptr = ptr.right;
}
}
int val = sw1.val;
sw1.val = sw2.val;
sw2.val = val;