2034-StockPriceFluctuation

Stock Price Fluctuation

Problem can be found in here!

Solution

class StockPrice:
    def __init__(self):
        self.latest_time = 0
        self.max_heap = []
        self.min_heap = []
        self.time_price_map = dict()

    def update(self, timestamp: int, price: int) -> None:
        self.time_price_map[timestamp] = price
        self.latest_time = max(self.latest_time, timestamp)
        
        heappush(self.min_heap, (price, timestamp))
        heappush(self.max_heap, (-price, timestamp))
        

    def current(self) -> int:
        return self.time_price_map[self.latest_time]

    def maximum(self) -> int:
        price, time = self.max_heap[0]
        
        while -price != self.time_price_map[time]:
            heappop(self.max_heap)
            price, time = self.max_heap[0]
            
        return -price

    def minimum(self) -> int:
        price, time = self.min_heap[0]
        
        while price != self.time_price_map[time]:
            heappop(self.min_heap)
            price, time = self.min_heap[0]
            
        return price

Time Complexity: , Space Complexity: