每周一道Leetcode—— 501. Find Mode in Binary Search Tree

题目

给定一个BST二分查找树,求众数。
要求,除去递归的空间占用,其他代码空间复杂度为O(1)。

分析

很显然,二分查找树的中序遍历就是一个有序数组。
有序数组查找最长的连续重复的数字,就是众数了。需要注意的是,题目要求返回重复的数字。另外,空间复杂度必须为O(1)。要写出这样的代码并不困难。

运行16ms,击败62%

/**
 * 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> mode;
    int last;
    int lastCount;
    int modeCount;
public:
    vector<int> findMode(TreeNode* root) {
        last = 0;
        lastCount = 0;
        modeCount = 0;
        mode.clear();
        runMode(root);
        return mode;
    }

    void runMode(TreeNode* root){
        if(root != NULL){
            runMode(root -> left);
            int nowValue = root->val;
            int nowCount = nowValue != last? 1 : lastCount + 1;
            if(modeCount == nowCount){
                mode.push_back(nowValue);
            }
            if(modeCount < nowCount){
                modeCount = nowCount;
                mode.clear();
                mode.push_back(nowValue);
            }
            last = nowValue;
            lastCount = nowCount;
            runMode(root -> right);
        }
    }
    
};
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,769评论 0 33
  • LeetCode 刷题随手记 - 第一部分 前 256 题(非会员),仅算法题,的吐槽 https://leetc...
    蕾娜漢默阅读 17,911评论 2 36
  • 原题 给出一个所有元素以升序排序的单链表,将它转换成一棵高度平衡的二分查找树 解题思路 跟二叉树相关 - 分治法,...
    Jason_Yuan阅读 654评论 0 0
  • 原文出处:http://www.cnblogs.com/maybe2030/p/4715035.html引文出处:...
    明教de教主阅读 9,218评论 0 7
  • 白杨树, 你是树中的真君子, 你是树中的伟丈夫, 你有一种顶天立地的气节。 一生只有一个使命, 那就是向上,向上,...
    妙心阅读 777评论 5 3