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

最新推荐文章于 2024-06-14 05:45:37 发布

JehanRio

最新推荐文章于 2024-06-14 05:45:37 发布

阅读量54

点赞数 1

分类专栏： Leetcode专栏文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_51322383/article/details/131126945

版权

Leetcode专栏专栏收录该内容

278 篇文章 0 订阅

订阅专栏

文章目录

题目
代码（首刷看解析）

题目

在这里插入图片描述
Leetcode 501. 二叉搜索树中的众数

代码（首刷看解析）

class Solution {
public:
    vector<int> res;
    TreeNode* prev;
    vector<int> findMode(TreeNode* root) {
        int cur = 0, count = 0;
        traverse(root, cur, count);
        return res;
    }
    void traverse(TreeNode* root, int& cur, int& count) {
        if(!root)
            return;
        traverse(root->left, cur, count);
        if(!prev) {
            prev = root;   // 找到第一个结点
            cur = 1;
            count = 1;
            res.push_back(root->val);
        }
        else {
            if(root->val == prev->val) {
                cur++;
                if(cur == count) {
                    res.push_back(root->val);
                } else if(cur > count) {
                    res.clear();
                    count = cur;
                    res.push_back(root->val);
                }
            } else {
                cur = 1;
                if(cur == count) {
                    res.push_back(root->val);
                }
            }
        }
        prev = root;
        traverse(root->right, cur, count);
    }
};