二叉树系列---recover_binary_search_tree

题

目

在一个二叉查找树中，有两个节点被错误的交换了，实现一个函数得出正确的二叉查找树；

思路

利用二叉查找树性质：中序遍历有序；
*先得到中序序列；
* 正向变量中序序列，遇到逆序的就break；
* 反向遍历中序序列，遇到逆序的就break；
* 交换两个break处的元素即可；

实现

public void recoverTree(TreeNode root){
        ArrayList<TreeNode> list=new ArrayList<TreeNode>();
        inOrder(list,root);

        TreeNode l=null,r=null;
        //正向变量
        for(int i=0;i<list.size()-1;i++){
            if(list.get(i).val>list.get(i+1).val){
                l=list.get(i);
                break;
            }
        }
        //反向变量
        for(int i=list.size()-1;i>=1;i--){
            if(list.get(i).val<list.get(i-1).val){
                r=list.get(i);
                break;
            }
        }

        //交换
        int tmp=l.val;
        l.val=r.val;
        r.val=tmp;
    }

    private void inOrder(ArrayList<TreeNode> inOrderList, TreeNode root) {
        if(root==null){
            return;
        }
        inOrder(inOrderList,root.left);
        inOrderList.add(root);
        inOrder(inOrderList,root.right);
    }