【LeetCode】设计有序流 Java 题解

作者：Albert

2022 年 8 月 16 日
北京
本文字数：1275 字
阅读完需：约 4 分钟

题目描述

有 n 个 (id, value) 对，其中 id 是 1 到 n 之间的一个整数，value 是一个字符串。不存在 id 相同的两个 (id, value) 对。

设计一个流，以任意顺序获取 n 个 (id, value) 对，并在多次调用时按 id 递增的顺序返回一些值。

实现 OrderedStream 类：

OrderedStream(int n) 构造一个能接收 n 个值的流，并将当前指针 ptr 设为 1 。String[] insert(int id, String value) 向流中存储新的 (id, value) 对。存储后：如果流存储有 id = ptr 的 (id, value) 对，则找出从 id = ptr 开始的最长 id 连续递增序列，并按顺序返回与这些 id 关联的值的列表。然后，将 ptr 更新为最后那个 id + 1 。否则，返回一个空列表。

输入["OrderedStream", "insert", "insert", "insert", "insert", "insert"][[5], [3, "ccccc"], [1, "aaaaa"], [2, "bbbbb"], [5, "eeeee"], [4, "ddddd"]]输出[null, [], ["aaaaa"], ["bbbbb", "ccccc"], [], ["ddddd", "eeeee"]]
解释OrderedStream os= new OrderedStream(5);os.insert(3, "ccccc"); // 插入 (3, "ccccc")，返回 []os.insert(1, "aaaaa"); // 插入 (1, "aaaaa")，返回 ["aaaaa"]os.insert(2, "bbbbb"); // 插入 (2, "bbbbb")，返回 ["bbbbb", "ccccc"]os.insert(5, "eeeee"); // 插入 (5, "eeeee")，返回 []os.insert(4, "ddddd"); // 插入 (4, "ddddd")，返回 ["ddddd", "eeeee"]
来源：力扣（LeetCode）链接：https://leetcode.cn/problems/design-an-ordered-stream著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

复制代码

思路分析

今天的算法题目是设计题目，题目要求设计一个流，可以满足以顺序获取 n 个 (id, value) 对，并将当前指针 ptr 设为 1。根据上述条件，我们可以设置 ptr 计数变量，由于 insert 方法定义了 insert(int id, String value)，传参列表，比较容易想到使用 hashMap 数据结构存储流的值。观察例子，这里面比较难理解的是添加一个参数之后，当 ptr == idKey 的时候，才进入循环，累加结果，同时 idKey++, ptr++。理解之后，代码就容易写了。具体实现代码如下，供参考。

通过代码

class OrderedStream {    Map<Integer, String> map;    int ptr;    public OrderedStream(int n) {        this.map = new HashMap<>(n);        this.ptr = 1;    }        public List<String> insert(int idKey, String value) {        List<String> ans = new LinkedList<>();        this.map.put(idKey, value);        if (this.ptr == idKey) {            while (this.map.containsKey(idKey)) {                ans.add(this.map.get(idKey));                idKey++;                this.ptr++;            }        }        return ans;    }}
/** * Your OrderedStream object will be instantiated and called as such: * OrderedStream obj = new OrderedStream(n); * List<String> param_1 = obj.insert(idKey,value); */