饭后小甜点leetcode——恢复二叉树

有一种题,需要在遍历二叉树的时候,keep一个last指针,始终指向上一个遍历到的节点。

恢复二叉树

恢复二叉搜索树

【思路】二叉搜索树在中序遍历的时候,得到的list应该如图中粉色点点那样递增,但如果其中有两个节点被放错地方了,则这两个节点则会像绿色点点一样,
在这里插入图片描述
所以在第一次发现last节点比current节点大的时候,如果first节点还没有赋值过,则说明是图中第一个圈中的情况,然后把last节点赋值给first,到了再一次发现last比current大的时候,而且first已经有值了,就是圈2中的情况,则此时second节点更新为current节点。

public class Solution {
    public TreeNode last;
    public TreeNode first;
    public TreeNode second;
    public void RecoverTree(TreeNode root)
    {
        DFS(root);
        var temp = first.val;
        first.val = second.val;
        second.val = temp;
    }
    
    public void DFS(TreeNode root){
        if(root==null){
            return;
        }
        DFS(root.left);
        if(last!=null&&last.val>root.val){
            if(first==null){
                first = last;
            }
            second = root;
        }
        last = root;
        DFS(root.right);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值