码迷,mamicode.com
首页 > 其他好文 > 详细

【LeetCode】Recover Binary Search Tree

时间:2014-05-21 19:11:16      阅读:266      评论:0      收藏:0      [点我收藏+]

标签:style   blog   class   c   code   java   

Two elements of a binary search tree (BST) are swapped by mistake.

Recover the tree without changing its structure.

Note:
A solution using O(n) space is pretty straight forward. Could you devise a constant space solution?

bubuko.com,布布扣
public class Solution {
    public void recoverTree(TreeNode root) {
        if(root!=null){
            Stack<TreeNode> stack = new Stack<TreeNode>();
            TreeNode cur = root;
            int min=Integer.MIN_VALUE;
            int preint = Integer.MIN_VALUE;
            TreeNode pre=root;
            TreeNode after=pre;
            boolean bol = true;
            while(!stack.isEmpty()||cur!=null){
                while(cur!=null){
                    stack.push(cur);
                    cur=cur.left;
                }
                if(!stack.isEmpty()){
                    int temp = stack.peek().val;
                    if(bol&&temp>preint){
                        preint = temp;
                        pre=stack.peek();
                    }
                    if(bol&&temp<preint){
                        bol=false;
                        min=temp;
                        after=stack.peek();
                        cur=stack.peek().right;
                        stack.pop();
                        continue;
                    }
                    if(!bol&&temp<min){
                        min=temp;
                        after=stack.peek();
                    }
                    cur=stack.peek().right;
                    stack.pop();
                    
                }
            }
            int tt  = pre.val;
            pre.val=after.val;
            after.val=tt;
        }
    }
}
bubuko.com,布布扣

 

【LeetCode】Recover Binary Search Tree,布布扣,bubuko.com

【LeetCode】Recover Binary Search Tree

标签:style   blog   class   c   code   java   

原文地址:http://www.cnblogs.com/yixianyixian/p/3740197.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!