-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathFrequencyTracker.cpp
More file actions
41 lines (36 loc) · 1.04 KB
/
FrequencyTracker.cpp
File metadata and controls
41 lines (36 loc) · 1.04 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
class FrequencyTracker {
public:
FrequencyTracker() {
}
void add(int number) {
int old_freq = num_to_freq[number];
num_to_freq[number]++;
if (old_freq) {
freq_to_nums[old_freq].erase(number);
}
freq_to_nums[old_freq+1].insert(number);
}
void deleteOne(int number) {
int old_freq = num_to_freq[number];
if (old_freq) {
num_to_freq[number]--;
freq_to_nums[old_freq].erase(number);
if (--old_freq) {
freq_to_nums[old_freq].insert(number);
}
}
}
bool hasFrequency(int frequency) {
return (freq_to_nums[frequency].size());
}
private:
unordered_map<int,int> num_to_freq;
unordered_map<int,unordered_set<int>> freq_to_nums;
};
/**
* Your FrequencyTracker object will be instantiated and called as such:
* FrequencyTracker* obj = new FrequencyTracker();
* obj->add(number);
* obj->deleteOne(number);
* bool param_3 = obj->hasFrequency(frequency);
*/