其他分享
首页 > 其他分享> > 501. 二叉搜索树中的众数

501. 二叉搜索树中的众数

作者:互联网

力扣标记的难度的[简单],但看了题解并不简单,仔细一看题解是吧递归搜索变成了两层循环实现

力扣原题:501. 二叉搜索树中的众数

1. 题目描述

假定 BST 有如下定义:

结点左子树中所含结点的值小于等于当前结点的值
结点右子树中所含结点的值大于等于当前结点的值
左子树和右子树都是二叉搜索树
例如:

给定 BST [1,null,2,2],

   1
    \
     2
    /
   2
返回[2].

提示:如果众数超过1个,不需考虑输出顺序

进阶:你可以不使用额外的空间吗?(假设由递归产生的隐式调用栈的开销不被计算在内)

2. 分析

第一想法:使用字典(键-值)数组保存所有扫描结果,之后进行一遍扫描,但这样费时费力;

第二想法:定义一个结构体,保存键-值(-- -- -_-换汤不换药),然后定义一个变量记录众数的最大个数,用以更新结构体数组;

之后的之后----结构体都不要了,直接变量表示。

3. 代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
private:
    vector<int> answer;
    int maxCount,base,count;
public:
    void update(int cur){
        if(cur == base){
            count++;
        }
        else{
            count = 1;
            base = cur;
        }
        if(count == maxCount){
            answer.push_back(cur);
        }
        else if(count > maxCount){
            maxCount = count;
            answer = vector<int>{base};//非常好用的STL
        }
    }
    // 中序遍历
    void dfs(TreeNode* root){
        if(root == NULL){
            return;
        }
        dfs(root->left);
        update(root->val);
        dfs(root->right);
    }

    vector<int> findMode(TreeNode* root) {
        if(root == NULL){
            return vector<int>{};
        }
        base = root->val;
        count = 0;
        dfs(root);
        return answer;
    }
};

标签:count,结点,TreeNode,树中,base,众数,answer,root,501
来源: https://www.cnblogs.com/libbin/p/find-mode-in-binary-search-tree.html