leet-code/src/main/java/leetcode/editor/cn/StockPriceFluctuation.java

//给你一支股票价格的数据流。数据流中每一条记录包含一个 时间戳 和该时间点股票对应的 价格 。 
//
// 不巧的是，由于股票市场内在的波动性，股票价格记录可能不是按时间顺序到来的。某些情况下，有的记录可能是错的。如果两个有相同时间戳的记录出现在数据流中，前一条
//记录视为错误记录，后出现的记录 更正 前一条错误的记录。 
//
// 请你设计一个算法，实现： 
//
// 
// 更新 股票在某一时间戳的股票价格，如果有之前同一时间戳的价格，这一操作将 更正 之前的错误价格。 
// 找到当前记录里 最新股票价格 。最新股票价格 定义为时间戳最晚的股票价格。 
// 找到当前记录里股票的 最高价格 。 
// 找到当前记录里股票的 最低价格 。 
// 
//
// 请你实现 StockPrice 类： 
//
// 
// StockPrice() 初始化对象，当前无股票价格记录。 
// void update(int timestamp, int price) 在时间点 timestamp 更新股票价格为 price 。 
// int current() 返回股票 最新价格 。 
// int maximum() 返回股票 最高价格 。 
// int minimum() 返回股票 最低价格 。 
// 
//
// 
//
// 示例 1： 
//
// 输入：
//["StockPrice", "update", "update", "current", "maximum", "update", "maximum", 
//"update", "minimum"]
//[[], [1, 10], [2, 5], [], [], [1, 3], [], [4, 2], []]
//输出：
//[null, null, null, 5, 10, null, 5, null, 2]
//
//解释：
//StockPrice stockPrice = new StockPrice();
//stockPrice.update(1, 10); // 时间戳为 [1] ，对应的股票价格为 [10] 。
//stockPrice.update(2, 5);  // 时间戳为 [1,2] ，对应的股票价格为 [10,5] 。
//stockPrice.current();     // 返回 5 ，最新时间戳为 2 ，对应价格为 5 。
//stockPrice.maximum();     // 返回 10 ，最高价格的时间戳为 1 ，价格为 10 。
//stockPrice.update(1, 3);  // 之前时间戳为 1 的价格错误，价格更新为 3 。
//                          // 时间戳为 [1,2] ，对应股票价格为 [3,5] 。
//stockPrice.maximum();     // 返回 5 ，更正后最高价格为 5 。
//stockPrice.update(4, 2);  // 时间戳为 [1,2,4] ，对应价格为 [3,5,2] 。
//stockPrice.minimum();     // 返回 2 ，最低价格时间戳为 4 ，价格为 2 。
// 
//
// 
//
// 提示： 
//
// 
// 1 <= timestamp, price <= 10⁹ 
// update，current，maximum 和 minimum 总 调用次数不超过 10⁵ 。 
// current，maximum 和 minimum 被调用时，update 操作 至少 已经被调用过 一次 。 
// 
// 👍 6 👎 0

package leetcode.editor.cn;

import java.util.TreeMap;

//2034:股票价格波动
class StockPriceFluctuation {
    public static void main(String[] args) {
        //测试代码
//        Solution solution = new StockPriceFluctuation().new Solution();
        StockPrice stockPrice = new StockPriceFluctuation().new StockPrice();
        stockPrice.update(38, 2308);
        stockPrice.maximum();
        stockPrice.current();
        stockPrice.minimum();
        stockPrice.maximum();
        stockPrice.maximum();
        stockPrice.maximum();
        stockPrice.minimum();
        stockPrice.minimum();
        stockPrice.maximum();
        stockPrice.update(47, 7876);
        stockPrice.maximum();
        stockPrice.minimum();
        stockPrice.update(58, 1866);
        stockPrice.maximum();
        stockPrice.minimum();
        stockPrice.current();
        stockPrice.maximum();
        stockPrice.update(43, 121);
        stockPrice.minimum();
        stockPrice.maximum();
        stockPrice.update(40, 5339);
        stockPrice.maximum();
        stockPrice.maximum();
        stockPrice.current();
        stockPrice.update(32, 5339);
        stockPrice.current();//
        stockPrice.minimum();
        stockPrice.update(43, 6414);
        stockPrice.update(49, 9369);
        stockPrice.minimum();
        stockPrice.minimum();
        stockPrice.update(36, 3192);
        stockPrice.current();//
        stockPrice.update(48, 1006);
        stockPrice.maximum();
        stockPrice.update(53, 8013);
        stockPrice.minimum();
    }

    //力扣代码
    //leetcode submit region begin(Prohibit modification and deletion)
    class StockPrice {

        TreeMap<Integer, Integer> map;
        TreeMap<Integer, Integer> priceMap;
        int cur;

        public StockPrice() {
            map = new TreeMap<>();
            priceMap = new TreeMap<>();
            cur = 0;
        }

        public void update(int timestamp, int price) {
            if (map.containsKey(timestamp)) {
                int oldPrice = map.get(timestamp);
                priceMap.put(oldPrice, priceMap.get(oldPrice) - 1);
                if (priceMap.get(oldPrice) == 0) {
                    priceMap.remove(oldPrice);
                }
            }
            priceMap.put(price, priceMap.getOrDefault(price, 0) + 1);
            map.put(timestamp, price);
            cur = Math.max(cur, timestamp);
        }

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

        public int maximum() {
            return priceMap.lastKey();
        }

        public int minimum() {
            return priceMap.firstKey();
        }
    }

/**
 * 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();
 */
//leetcode submit region end(Prohibit modification and deletion)

}