99. 恢复二叉搜索树

99. 恢复二叉搜索树

二叉搜索树中的两个节点被错误地交换。

请在不改变其结构的情况下,恢复这棵树。

示例 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) 空间复杂度的解法很容易实现。
  • 你能想出一个只使用常数空间的解决方案吗?

不好意思我想不出常数空间的,只能去看题解了。自己写的是O(N)的。感觉也还可以吧

class Solution {
    List<Integer> list = new ArrayList<>();

    public void inorder(TreeNode root){
        if(root == null)
            return;
        inorder(root.left);
        list.add(root.val);
        inorder(root.right);
    }
    public void recover(TreeNode root,int x ,int y){
        if(root == null)
            return;
        if(root.val == x || root.val == y){
            if(root.val == x)
                root.val = y;
            else
                root.val = x;
        }
        recover(root.left, x , y);
        recover(root.right, x , y);
    }
    public void recoverTree(TreeNode root) {
        inorder(root);
        int x = 0, y = 0;
        int len = list.size();
        for(int i = 0 ; i < len-1 ; i++)
            if(list.get(i) > list.get(i+1)){
                x = list.get(i);
                break;
            }
        for(int i = len-1 ; i > 0 ; i--)
            if(list.get(i) < list.get(i-1)){
                y = list.get(i);
                break;
            }
        recover(root,x,y);
    }
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值