【LeetCode】二叉搜索树中的众数

最新推荐文章于 2022-01-30 09:24:43 发布

头号理想

最新推荐文章于 2022-01-30 09:24:43 发布

阅读量265

点赞数

文章标签：二叉树数据结构 leetcode 算法 hash

本文链接：https://blog.csdn.net/weixin_44302602/article/details/112280722

版权

算法专栏收录该内容

59 篇文章 0 订阅

订阅专栏

该博客讨论了一种优化方法来解决在二叉搜索树中找到众数的问题。原始解法通过遍历所有节点并使用哈希表和集合实现，但空间复杂度较高。优化后的解法利用了二叉搜索树的特性，在遍历过程中实时计数，减少了空间消耗，提高了效率。这种方法仅适用于二叉搜索树，能更有效地找到众数或众数集合。

摘要由CSDN通过智能技术生成

题目

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

假定 BST 有如下定义：

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

在这里插入图片描述

暴力数组哈希

看到这个题我首先想到的就是遍历二叉树然后将所有节点的值都放在一个驻足中
然后通过哈希表两次遍历数组得到众数（当然排序也可以用其他的办法）
这个方法的时间复杂度是O(N)
但是我这里不仅用到了哈希表还用到了set
所以空间复杂度消耗比较大

  vector<int> nums;
void def(TreeNode* root) {
	if (root == NULL) {
		return;
	}
	nums.push_back(root->val);
	def(root->left);
	def(root->right);
}
//添加完毕


vector<int> findMode(TreeNode* root) {
	def(root);
	map<int, int> hash;
	set<int> s;
	int maxs = 0;
	vector<int> ans;
	for (int i = 0; i < nums.size(); i++) {
		hash[nums[i]]++;
		maxs = max(maxs, hash[nums[i]]);
	}
	for (int i = 0; i < nums.size(); i++) {
		if (hash[nums[i]] == maxs && !s.count(nums[i])) {
			ans.push_back(nums[i]);
			s.insert(nums[i]);
		}
	}
	return ans;
}

上边的解法有一个缺点，就是没有用到二叉搜索树的特性
上边的解法可用于所有二叉树
反正都是遍历一遍没什么区别
在这里插入图片描述

优化之后的解法

我们根据二叉搜索树的性质，我们边遍历边计数
从而得到频率最高的数也就是众数或者是众数集合

class Solution {
private:
    int count;
    int maxCount;
    TreeNode* pre;
    vector<int> result;
    void searchBST(TreeNode* cur) {
        if (cur == NULL) return ;

        searchBST(cur->left);       // 左
                                    // 中
        if (pre == NULL) { // 第一个节点
            count = 1;
        } else if (pre->val == cur->val) { // 与前一个节点数值相同
            count++;
        } else { // 与前一个节点数值不同
            count = 1;
        }
        pre = cur; // 更新上一个节点

        if (count == maxCount) { // 如果和最大值相同，放进result中
            result.push_back(cur->val);
        }

        if (count > maxCount) { // 如果计数大于最大值
            maxCount = count;
            result.clear();     // 很关键的一步，不要忘记清空result，之前result里的元素都失效了
            result.push_back(cur->val);
        }

        searchBST(cur->right);      // 右
        return ;
    }

public:
    vector<int> findMode(TreeNode* root) {
        int count = 0; // 记录元素出现次数
        int maxCount = 0;
        TreeNode* pre = NULL; // 记录前一个节点
        result.clear();

        searchBST(root);
        return result;
    }
};