-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathm2671.py
30 lines (23 loc) · 1.03 KB
/
m2671.py
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
class FrequencyTracker:
def __init__(self):
self.valToFrequency = defaultdict(int)
self.frequencyFrequencies = defaultdict(int)
def add(self, number: int) -> None:
if self.valToFrequency[number] != 0 :
self.frequencyFrequencies[self.valToFrequency[number]] -= 1
self.valToFrequency[number] += 1
self.frequencyFrequencies[self.valToFrequency[number]] += 1
def deleteOne(self, number: int) -> None:
if self.valToFrequency[number] <= 0 :
return
self.frequencyFrequencies[self.valToFrequency[number]] -= 1
self.valToFrequency[number] -= 1
if self.valToFrequency[number] > 0 :
self.frequencyFrequencies[self.valToFrequency[number]] += 1
def hasFrequency(self, frequency: int) -> bool:
return self.frequencyFrequencies[frequency] > 0
# Your FrequencyTracker object will be instantiated and called as such:
# obj = FrequencyTracker()
# obj.add(number)
# obj.deleteOne(number)
# param_3 = obj.hasFrequency(frequency)