wrong_ans_prob_3.java

gistfile1.txt

class StockPrice {
    
    TreeMap<Integer,Integer> timeToPrice;
    TreeMap<Integer,Integer> priceToTime;
    int current;

    public StockPrice() {
        timeToPrice = new TreeMap<>();   
        priceToTime = new TreeMap<>();
        current =0;
    }
    
    public void update(int timestamp, int price) {
        
        if (timeToPrice.containsKey(timestamp)) {
            priceToTime.remove(timeToPrice.get(timestamp));
        } else {
            current = price;
        }
        timeToPrice.put(timestamp, price);          
        priceToTime.put(price, timestamp);
        
    }
    
    public int current() {
        return current;
    }
    
    public int maximum() {
        return priceToTime.lastKey();
    }
    
    public int minimum() {
        return priceToTime.firstKey();
    }
}

/**
                 [1, 10]
["StockPrice", "update", "update", "current", "maximum", "update", "maximum", "update", "minimum"]
[null,          null, null, 5, 10, null, 5, null, 2]
[null,          null, null, 5, 1,  null, 1,  null,  4]


 * Your StockPrice object will be instantiated and called as such:
 * StockPrice obj = new StockPrice();
 * obj.update(timestamp,price);
 * int param_2 = obj.current();
 * int param_3 = obj.maximum();
 * int param_4 = obj.minimum();
 */