Leetcode99. 恢复二叉搜索树

Posted 2022-09-03 Java全栈研发大联盟

题目传送：​​https://leetcode.cn/problems/recover-binary-search-tree/​​

运行效果

代码如下

class Solution 
    //用于存储中序遍历的结果
  List<Integer> list = new ArrayList();

  public void recoverTree(TreeNode root) 
    // 1、先对这颗二叉树进行中序遍历，然后依次存放进list中
    inOrder(root);
    // m和n用于记录需要交换的2个位置
    int m = -1;
    int n = -1;

    for (int i = 0; i < list.size() - 1; i++) 
      // 第一种情况  非相邻交换： 1 2 3 4 5 6 7   交换 2和6的位置    变成1 6 3 4 5 2 7
      // 第二种情况  相邻交换： 1 2 3 4 5 6 7 交换2和3的2位置  变成  1 3 2 4 5 6 7
      if (list.get(i) > list.get(i + 1)) 
        if (m == -1) 
          m = i;
          continue;
        
        if (n == -1) 
          n = i + 1;
        
      
    
    // 如果m和n  都不等于-1，说明是属于第一种情况
    if (m != -1 && n != -1) 
      // 交换m和n位置的元素即可
      Integer val1 = list.get(m);
      Integer val2 = list.get(n);
      swapNode(val1,val2,root);
    
    // 如果m != -1   ，但是n== -1   说明是属于第二种情况
    if (m != -1 && n == -1) 
      // 交换m和m+1位置的元素即可
      Integer val1 = list.get(m);
      Integer val2 = list.get(m+1);
      swapNode(val1,val2,root);
    
  
  /*
   * @description: 在二叉树中交换 i和j位置的元素
   * @date: 2022/8/27 16:38
   * @param: a
   * @return: a
   */
  public void swapNode(int val1, int val2, TreeNode root) 
    if (root == null) 
      return;
    
    // 1、先遍历左子树
    swapNode(val1,val2,root.left);
    // 2、遍历当前根节点
    if (root.val == val1) 
      root.val = val2;
     else if (root.val == val2) 
      root.val = val1;
    
    // 3、最后遍历右子树
    swapNode(val1,val2,root.right);
  

  /*
   * @description: 中序遍历二叉树
   * @date: 2022/8/27 16:19
   * @param: a
   * @return: a
   */
  public void inOrder(TreeNode root) 
    if (root == null) 
      return;
    
    // 1、先遍历左子树
    inOrder(root.left);
    // 2、遍历当前根节点
    list.add(root.val);
    // 3、最后遍历右子树
    inOrder(root.right);