g2001_2100.s2034_stock_price_fluctuation.StockPrice Maven / Gradle / Ivy

Go to download
package g2001_2100.s2034_stock_price_fluctuation;

// #Medium #Hash_Table #Design #Heap_Priority_Queue #Ordered_Set #Data_Stream
// #2022_05_25_Time_163_ms_(65.51%)_Space_155.8_MB_(37.90%)

import java.util.HashMap;
import java.util.Map;
import java.util.PriorityQueue;

public class StockPrice {
    private static class Record {
        int time;
        int price;

        Record(int time, int price) {
            this.time = time;
            this.price = price;
        }
    }

    private Map map;
    private PriorityQueue maxHeap;
    private PriorityQueue minHeap;
    private int latestTimestamp = 0;

    public StockPrice() {
        map = new HashMap<>();
        maxHeap = new PriorityQueue<>((r1, r2) -> Integer.compare(r2.price, r1.price));
        minHeap = new PriorityQueue<>((r1, r2) -> Integer.compare(r1.price, r2.price));
    }

    public void update(int timestamp, int price) {
        latestTimestamp = Math.max(timestamp, latestTimestamp);
        maxHeap.offer(new Record(timestamp, price));
        minHeap.offer(new Record(timestamp, price));
        map.put(timestamp, price);
    }

    public int current() {
        return map.get(latestTimestamp);
    }

    public int maximum() {
        while (true) {
            Record rec = maxHeap.peek();
            if (map.get(rec.time).equals(rec.price)) {
                return rec.price;
            }
            maxHeap.poll();
        }
    }

    public int minimum() {
        while (true) {
            Record rec = minHeap.peek();
            if (map.get(rec.time).equals(rec.price)) {
                return rec.price;
            }
            minHeap.poll();
        }
    }
}