题目:一棵二叉树原本是搜索二叉树,但是其中有两个节点调换了位置,使得这棵二叉树不再是搜索二叉树,请找到这两个错误节点并返回他们的值。保证二叉树中结点的值各不相同。
给定一棵树的根结点,请返回两个调换了位置的值,其中小的值在前。
Java代码:
import java.util.*;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}*/
public class FindErrorNode {
public int[] findError(TreeNode node) {
/*
二叉搜索树的中序遍历,依次出现的值会一直升序,如果两个节点值错了,那么会出现降序。
第一个错误节点为第一次降序时较大的节点,第二个错误节点是最后一次降序时较小的节点。
该程序是一个改写的中序遍历。(本文用非递归模式)
*/
int[] res=new int[2];
TreeNode[] err=new TreeNode[2];
Stack<TreeNode> st=new Stack<TreeNode>();
TreeNode cur=node;
TreeNode point=null;//用来比较的第一个元素
while(!st.isEmpty()||cur!=null)
{
if(cur!=null)
{
st.push(cur);
cur=cur.left;
}else
{
cur=st.pop();
if(point!=null&&point.val>cur.val)
{
err[0]=err[0]==null?point:err[0];//err[0]是空的,值为point,否则值不变
err[1]=cur;
}
point=cur;
cur=cur.right;//cur.right存在cur=cur.right,否则cur为null。开始下一次的循环。
}
}
res[1]=err[0].val;
res[0]=err[1].val;
return res;
}
}