leetcode 恢复二叉搜索树

二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下，恢复这棵树。

示例 1:

输入: [1,3,null,null,2]

   1
  /
 3
  \
   2

输出: [3,1,null,null,2]

   3
  /
 1
  \
   2

示例 2:

输入: [3,1,4,null,null,2]

  3
 / \
1   4
   /
  2

输出: [2,1,4,null,null,3]

  2
 / \
1   4
   /
  3

进阶:

使用 O(n) 空间复杂度的解法很容易实现。
你能想出一个只使用常数空间的解决方案吗？

思路：因为二叉搜索树的中序遍历应该是递增的，所以可以从中序遍历上入手，先得出二叉树的中序遍历，存在一个list里面，然后找出list里不是递增的两个元素，最后再遍历一遍二叉树，将错误的元素值改正过来

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left;
 *     public TreeNode right;
 *     public TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
	//得到中序遍历
    public void Inordertravalsal(TreeNode root,IList<int> list)
    {
        if(root!=null)
        {
            if(root.left!=null)
                Inordertravalsal(root.left,list);
            list.Add(root.val);
            if(root.right!=null)
                Inordertravalsal(root.right,list);   
        }
    }
	//再次遍历并在遇到错误元素时改正元素
    public void swap(TreeNode root,int[] pos)
    {
        if(root!=null)
        {
            if(root.val==pos[0])
                root.val=pos[1];
            else if(root.val==pos[1])
                root.val=pos[0];
            
            if(root.left!=null)
                swap(root.left,pos);
            
            if(root.right!=null)
                swap(root.right,pos);
            
               
        }
    }

    public void RecoverTree(TreeNode root) 
    {
        IList<int> list=new List<int>();
        int i,x=-1,y=-1;
        Inordertravalsal(root,list);
        //找出错误的元素
        for(i=0;i<list.Count-1;i++)
        {
            if(list[i]>list[i+1])
            {
                y=list[i+1];
                if(x==-1)
                x=list[i];
            }
        } 
        int[] temp=new int[2]{x,y};      
        swap(root,temp);

    }
}

时间复杂度：O（n）空间复杂度：O（n）

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/recover-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。