LeetCode 6417.频率跟踪器

208 阅读2分钟

从零开始学习c++,每天起码做一道leetcode题目,在此记录,希望最后能够有所收获!

一、题目描述

6417.频率跟踪器

请你设计并实现一个能够对其中的值进行跟踪的数据结构,并支持对频率相关查询进行应答。

实现 FrequencyTracker 类:

  • FrequencyTracker():使用一个空数组初始化 FrequencyTracker 对象。
  • void add(int number):添加一个 number 到数据结构中。
  • void deleteOne(int number):从数据结构中删除一个 number 。数据结构 可能不包含 number ,在这种情况下不删除任何内容。
  • bool hasFrequency(int frequency): 如果数据结构中存在出现 frequency 次的数字,则返回 true,否则返回 false。  

示例 1:

输入
["FrequencyTracker", "add", "add", "hasFrequency"]
[[], [3], [3], [2]]
输出
[null, null, null, true]

解释
FrequencyTracker frequencyTracker = new FrequencyTracker();
frequencyTracker.add(3); // 数据结构现在包含 [3]
frequencyTracker.add(3); // 数据结构现在包含 [3, 3]
frequencyTracker.hasFrequency(2); // 返回 true ,因为 3 出现 2 次

示例 2:

输入
["FrequencyTracker", "add", "deleteOne", "hasFrequency"]
[[], [1], [1], [1]]
输出
[null, null, null, false]

解释
FrequencyTracker frequencyTracker = new FrequencyTracker();
frequencyTracker.add(1); // 数据结构现在包含 [1]
frequencyTracker.deleteOne(1); // 数据结构现在为空 []
frequencyTracker.hasFrequency(1); // 返回 false ,因为数据结构为空

示例 3:

输入
["FrequencyTracker", "hasFrequency", "add", "hasFrequency"]
[[], [2], [3], [1]]
输出
[null, false, null, true]

解释
FrequencyTracker frequencyTracker = new FrequencyTracker();
frequencyTracker.hasFrequency(2); // 返回 false ,因为数据结构为空
frequencyTracker.add(3); // 数据结构现在包含 [3]
frequencyTracker.hasFrequency(1); // 返回 true ,因为 3 出现 1 次

二、思路分析

有想法,可惜最后超时了。想了半天也没想出来该怎么优化。

最好的解法就是用两个map,一个记录每个数的出现次数,另一个记录出现次数的出现次数,这样在最后一个函数我们就能通过第二个map直接得到答案。

其实类似于套娃,多做就好了。

三、AC代码

class FrequencyTracker {
    unordered_map<int, int> cnt; // 每个数的出现次数
    unordered_map<int, int> freq; // 出现次数的出现次数

public:
    FrequencyTracker() {
    } 
    void add(int number) {
        --freq[cnt[number]];
        cnt[number]++;
        freq[cnt[number]]++;
    }   
    void deleteOne(int number) {
        if(cnt[number]){
            --freq[cnt[number]];
            cnt[number]--;
            freq[cnt[number]]++;
        }
    } 
    bool hasFrequency(int frequency) {
        return freq[frequency];
    }
};

四、总结

没遇到过这种题。