二叉树系列---recover_binary_search_tree

来源:互联网 发布:2001年全明星赛数据 编辑:程序博客网 时间:2024/06/05 16:24

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

思路

利用二叉查找树性质:中序遍历有序;
*先得到中序序列;
* 正向变量中序序列,遇到逆序的就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);    }
0 0
原创粉丝点击