深度优先搜索的中序遍历:
1.应为只有两个值是错误的,所以我们可以用中序遍历来进行遍历这个二叉搜索数,二叉搜索树的中序遍历是一个逐级递增的数组,如果当前节点比前一个节点小,那么其中有一个节点肯定是错误了,只需要找出这两个错误的节点,再交换他们的值。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public void recoverTree(TreeNode root) {
//定义一个站来模拟中序遍历的进栈和出栈过程
Stack<TreeNode> st = new Stack<>();
TreeNode i = null;//当前节点的前一个节点
TreeNode j = null;//第一个错误的节点
TreeNode k = null;//第二个错误的节点
while (root!= null || !st.isEmpty()){
//遍历的过程和遍历的方向
if (root!= null){
//节点进栈
st.push(root);
//先便利数的左节点
root = root.left;
}else{
//遍历完全部的左节点后,来出栈该节点,同时进行给节点的右节点进站。
root = st.pop();
//访问当前节点的一些逻辑操作
if (j== null && i!= null && i.val > root.val) {
j=i;
}
if (j!= null && i.val > root.val) {
k= root;
}
i= root;//更新i节点
root = root.right;
}
}
//交换两个节点的值
int tmp = j.val;
j.val=k.val;
k.val=tmp;
}
}