1.题目描述
有 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"]
2.解题思路与代码
2.1 解题思路
根据题目要求可知,插入流中的数据 idKey 从 1 到 n 共 n 个,并且只调用 n 次 insert 方法,每次调用的 idKey 不同,那么在整个数据流中调用的 insert 不会出现重复的 idKey,数据对是唯一的。根据这一特性我们就可以使用一个数组来存放插入的数据,数组的角标对应插入的 idKey-1 的值(因为 idKey 是从 1 到 n,数组角标从 0 到 n-1),数组元素为数据对的 value 字符串。并且使用一个变量 ptr 在数组上进行移动,从0开始。当 ptr 所指位置上的元素不为 nul 时,将 ptr 上的元素放入结果列表中,并且 ptr 右移一位,直到 ptr 超过数组长度或者 ptr 所指位置元素为 null 时停止。
2.2 代码
class OrderedStream {
private String[] arr;
int ptr = 0;
public OrderedStream(int n) {
arr = new String[n];
}
public List<String> insert(int idKey, String value) {
List<String> ans = new ArrayList<>();
arr[idKey - 1] = value;
while (ptr < arr.length && arr[ptr] != null) {
ans.add(arr[ptr++]);
}
return ans;
}
}
2.3 测试结果
通过测试
3.总结
- 使用数组存放每次插入数据对,idKey-1 为对应数组的位置
- 使用 ptr 在数组上移动,ptr 位置上元素部位 null 时将元素添加到结果列表并 ptr 右移