力扣:99. 恢复二叉搜索树

深度优先搜索的中序遍历:

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 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;
}
}

你可能感兴趣的:(树,leetcode,算法)