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

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

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

思路:

寻找二叉搜索树的最小绝对差,可以合理利用二叉树的特性,也就是中序遍历二叉搜索树的时候,其结果是一个升序数组,因为二叉搜索树的节点大小关系为左节点小于中间节点,右节点大于中间节点,所以按照左->中->右的顺序遍历即可得到一个升序数组。

利用这个特性,很容易想到的一个办法就是先中序遍历二叉树得到一个数组,再遍历这个数组并记录下来每相邻的两个元素的最小差值。但实际上根本没必要这么做,因为在遍历二叉树的时候就可以完成比较的工作,用一个变量记录最小绝对差,然后每次比较的时候都更新这个变量。

这个办法就是双指针法,为了每次遍历节点的时候都能和前一个节点计算差值,需要一个额外的指针记录前一个节点,在中序遍历的时候,我们每次都先处理prev和cur节点,然后取两个节点的最小差值,再更新cur和prev节点,令prev=cur,cur=cur->right。

        // 处理中间节点
        // 取两个节点的最小差值
        if (prev != nullptr)
            minDiff = min(minDiff, cur->val - prev->val);
        // 令prev等于当前节点
        prev = cur;

这种双指针的办法可以使我们遍历二叉树的时候同时处理两个节点,一个为当前遍历的节点,一个为前一个遍历过的节点,这样可以节省遍历数组所花的时间和空间。 

代码:

class Solution {
public:
    int minDiff = INT_MAX;
    TreeNode* prev = nullptr;
    int getMinimumDifference(TreeNode* root) {
        inorder(root);
        return minDiff;
    }
    void inorder(TreeNode* cur)
    {
        if (cur == nullptr)
            return;
        
        // 中序遍历,先遍历左边的节点
        inorder(cur->left);
        // 处理中间节点
        // 取两个节点的最小差值
        if (prev != nullptr)
            minDiff = min(minDiff, cur->val - prev->val);
        // 令prev等于当前节点
        prev = cur;
        // 遍历右边的节点
        inorder(cur->right);
        return;
    }
};

LeetCode 501.二叉搜索树中的众数

链接:501.二叉搜索树中的众数

思路:

注意二叉搜素数一般是没有重复节点的,但是这道题中的二叉树的元素可以重复。这道题的思路和530. 二叉搜索树的最小绝对差一致,都是用中序遍历的方法,然后用一个prev节点储存之前的节点方便和当前节点作比较。众数为二叉树中出现次数最多的数,因此在中序遍历的过程中,如果一个数出现了很多次,那么prev和cur节点的值一定会出现很多次相同的情况,因此只需要把prev和cur相同的情况的次数count记录下来,就知道这个数重复出现了多少次。

当prev和cur值相同时,count++,当不同时,count就重置为1,表示这个数只出现过一次。同时,我们需要另一个变量maxCount用于记录当前重复最多次数的数的重复次数是多少。众数可以是很多个,因此当发现了maxCount等于count的时候,就把当前数push到结果里,如果发现了maxCount小于count,则说明当前的数重复次数多于之前记录的次数最大值,那么当前的数便成为了新的众数,需要把之前记录在结果里的数全部clear,再把新的众数放进结果里。

代码:

class Solution {
public:
    int count = 1;
    vector<int> ans;
    TreeNode *prev;
    int maxCount = 1;
    vector<int> findMode(TreeNode* root) {
        inorder(root);
        return ans;
    }
    void inorder(TreeNode* root)
    {
        if (root == nullptr)
            return;
        inorder(root->left);
        if (prev != nullptr && prev->val == root->val)
            // 如果现节点和之前节点值一样,数量加1
            count++;
        else
            // 如果不一样,重置数量
            count = 1;
        if (maxCount == count)
            // 如果当前节点出现频率等于最大频率,说明有多个众数
            ans.push_back(root->val);
        else if (maxCount < count)
        {
            // 如果当前元素出现频率大于最大频率,更新最大频率
            maxCount = count;
            // 清除数组里的元素
            ans.clear();
            // 把新的最大频率的元素添加进来
            ans.push_back(root->val);
        }
        // 更新之前的节点
        prev = root;
        inorder(root->right);
        return;
    }
};

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

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

思路:

这道题和235. 二叉搜索树的最近公共祖先类似,但是解题过程却大不相同。这里采用后序遍历的方法,从二叉树的底端自底向上查找,分别用left和right两个节点表示查找到的p、q两个节点或者分别包含p、q两个节点的子树的根节点,判断方法为当root找到了p或者q的其中一个节点,就把这个节点返回给上一个父节点,如果没找到,则返回null。这样就相当于把p、q两个节点向上传递,直到这两个节点分别填满了left和right两个节点,因为root节点为left和right的父节点,这就表示root节点也是p和q的公共祖先。

为什么root一定是p和q的最近公共祖先呢?可以这样想,当left或者right其中一个为null的时候,这就表示了,root作为left或者right节点的祖先,如果left不为null,那么left一定是由p或者q的任意一个节点传递上来的,所以root一定也是p或者q其中之一的祖先,但是这个时候right为null,那么root一定不是p或者q中的另一个的祖先,要同时满足root成为两者的祖先,一定是left和right同时不为null的情况,所以在刚好满足这种情况的时候,root为第一个p和q的公共祖先,也就是最近公共祖先。

代码:

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == NULL)
            return NULL;
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        if (root->val == p->val || root->val == q->val)
            return root;
        if (left != NULL && right != NULL)
            return root;
        else if (left == NULL && right != NULL)
            return right;
        else if (left != NULL && right == NULL)
            return left;
        else
            return NULL;
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值