1656. 设计有序流 #

问题描述#

有 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"]
提示：

1 <= n <= 1000

1 <= id <= n

value.length == 5

value 仅由小写字母组成

每次调用 insert 都会使用一个唯一的 id

恰好调用 n 次 insert

解题思路#

class OrderedStream:
    def __init__(self, n: int):
        self.n = n + 1
        self.ptr = 1
        self.stream = [""] * (n + 1)

    def insert(self, id: int, value: str) -> List[str]:
        self.stream[id] = value
        if id == self.ptr:
            while self.ptr < self.n and self.stream[self.ptr]:
                self.ptr += 1
        return self.stream[id:self.ptr]

测试数据

os = OrderedStream(5)

assert os.insert(3, "ccccc") == []
assert os.insert(1, "aaaaa") == ["aaaaa"]
assert os.insert(2, "bbbbb") == ["bbbbb", "ccccc"]
assert os.insert(5, "eeeee") == []
assert os.insert(4, "ddddd") == ["ddddd", "eeeee"]

print("PASS")

1656. 设计有序流#

问题描述#

解题思路#

1656. 设计有序流 #