刷题-力扣-501. 二叉搜索树中的众数

501. 二叉搜索树中的众数

题目链接

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-mode-in-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目描述

给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。

假定 BST 有如下定义:

结点左子树中所含结点的值小于等于当前结点的值
结点右子树中所含结点的值大于等于当前结点的值
左子树和右子树都是二叉搜索树
例如:
给定 BST [1,null,2,2],

   1
    \
     2
    /
   2

返回[2].

提示:如果众数超过1个,不需考虑输出顺序
进阶:你可以不使用额外的空间吗?(假设由递归产生的隐式调用栈的开销不被计算在内)

题目分析

  1. 根据题目描述计算二叉搜搜树中的众数
  2. 中序遍历二叉搜索树,遍历的结果按照非递减排序nums
  3. 一次遍历nums计算众数

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> findMode(TreeNode* root) {
        vector<int> nums;
        vector<int> res;
        int max = 0;
        int nowMax = 1;
        dfs(root, nums);
        if (nums.empty()) return res;
        for (int i = 1; i < nums.size(); ++i) {
            if (nums[i] == nums[i - 1]) {
                ++nowMax;
            } else {
                if (nowMax > max) {
                    res.clear();
                    res.emplace_back(nums[i - 1]);
                    max = nowMax;
                    nowMax = 1;
                } else if (nowMax == max) {
                    res.emplace_back(nums[i - 1]);
                    max = nowMax;
                    nowMax = 1;
                } else {
                    nowMax = 1;
                }
            }
        }
        if (nowMax > max) {
            res.clear();
            res.emplace_back(nums[nums.size() - 1]);
        } else if (nowMax == max) {
            res.emplace_back(nums[nums.size() - 1]);
        }
        return res;
    }

private:
    void dfs(TreeNode* root, vector<int>& nums) {
        if (!root) return;
        dfs(root->left, nums);
        nums.emplace_back(root->val);
        dfs(root->right, nums);
        return;
    }
};
posted @ 2021-08-29 16:17  韩亚光  阅读(27)  评论(0编辑  收藏  举报