代码随想录算法训练营|530.二叉搜索树的最小绝对差、501.二叉搜索树中的众数 、 236. 二叉树的最近公共祖先

530.二叉搜索树的最小绝对差

题目

参考文章

思路:这道题是计算相邻两个节点的最小差值(不要被题目的意思迷惑了喔),这里我们用类似双指针的方式进行,设置要给pre记录当前节点的上一个节点,初始值为null,设置一个result记录差值。首先传入root(即当前节点)如果为null,则直接返回,利用中序遍历的方式遍历二叉树,但中间节点处理的时候,如果pre不为null,则计算pre和root之间的差值和result比较去最小值,如果pre为null则不进入判断,然后把当前节点赋值给pre即可

代码:

class Solution {
    TreeNode pre;// 记录上一个遍历的结点
    int result = Integer.MAX_VALUE;
    public int getMinimumDifference(TreeNode root) {
       if(root==null)return 0;
       traversal(root);
       return result;
    }

    public void traversal(TreeNode root){
        if(root==null)return;
        //左
        traversal(root.left);
        //中
        if(pre!=null){
            result = Math.min(result,root.val-pre.val);
        }
        pre = root;
        //右
        traversal(root.right);
    }
}

501.二叉搜索树中的众数 

题目

参考文章

思路:其实这题也用了双指针的方法,但是也有了其他方面的改变。利用双指针,记录当前节点和前一个节点的值,进行频率的判断(count),从而得到maxCount,并把对应最大的count即maxCount的对应的节点值加入到resList中(并随时准备清空resList)。具体看代码注释

代码:

class Solution {
    ArrayList<Integer> resList;//用于存储结果集
    int maxCount;//用于存储出现数值的最大频率
    int count;//用于存储目前遇到的元素数值的频率
    TreeNode pre;//记录当前节点的前一个遍历的节点(相当于双指针)
    public int[] findMode(TreeNode root) {
        resList = new ArrayList<>();
        maxCount = 0;
        count = 0;
        pre = null;
        findMode1(root);
        int[] res = new int[resList.size()];
        for (int i = 0; i < resList.size(); i++) {
            res[i] = resList.get(i);
        }
        return res;
    }

      public void findMode1(TreeNode root) {
        if (root == null) {
            return;
        }

        //利用中序遍历
        findMode1(root.left);

        int rootValue = root.val;//当前节点的值
        // 计数
        if (pre == null || rootValue != pre.val) { //如果pre为null或当前节点的值不等于前一个节点的值时,重置count为1
            count = 1;
        } else {//如果当前节点的值大于前一个节点的值,这增加这个值出现的频率 count++
            count++;
        }
        // 更新结果以及maxCount
        //重点!!!!
        if (count > maxCount) {//如果当前节点值出现的频率大于了目前最大的maxCount频率,则更新maxCount,并重置(清空)resList里面的值,把当前节点的值加入到resList中
            resList.clear();
            resList.add(rootValue);
            maxCount = count;
        } else if (count == maxCount) {//如果当前节点的频率等于最大出现频率maxCount,则把当前节点的值加入到resList中(因为出现的众数不止一个的)
            resList.add(rootValue);
        }
        pre = root; //再把当前节点赋值给pre即可

        findMode1(root.right);
    }
}

 236. 二叉树的最近公共祖先

题目

参考文章

思路:涉及到回溯,从底向上的一定要考虑后序遍历。这题用到了后序遍历的方式,重要部分是在中间节点时的操作,因为涉及到回溯,所以进行 if else判断时,要一层层回溯。 

这个递归结束条件  if (root == null || root == p || root == q) 如下

递归结束条件,这个条件把所有情况都涵盖在里面了,因为无论你p q在根节点左右边还是只在一边,又或者p以q为父亲的情况,都是满足的,到下面后序遍历处理中间节点的情况的时候(left和right都不为空),直接返回root(即p q共同的祖先),其他如果是p q在同一边(即通一条子树下),在中间节点处理过程中也是直接返回left或right,这样等回溯到上一层节点时,操作也是一样的

代码:

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null || root == p || root == q) { // 递归结束条件,这个条件把所有情况都涵盖在里面了,因为无论你p q在根节点左右边还是只在一边,又或者p以q为父亲的情况,都是满足的,到下面后序遍历处理中间节点的情况的时候(left和right都不为空),直接返回root(即p q共同的祖先),其他如果是p q在同一边(即通一条子树下),在中间节点处理过程中也是直接返回left或right,这样等回溯到上一层节点时,操作也是一样的
            return root;
        }

        // 后序遍历
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);

        if(left == null && right == null) { // 若未找到节点 p 或 q
            return null;
        }else if(left == null && right != null) { // 若找到一个节点
            return right;
        }else if(left != null && right == null) { // 若找到一个节点
            return left;
        }else { // 若找到两个节点
            return root;
        }
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值