LeetCode 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”]

思路就是每次插入后检查ptr当前所指位置是否有元素,如果有,则返回该元素及该元素之后的连续元素,并增加ptr:

class OrderedStream {
    
    
public:
    OrderedStream(int n) {
    
    
        svec.resize(n + 1);
    }
    
    vector<string> insert(int id, string value) {
    
    
        vector<string> ret;

        svec[id] = value;

        while (1) {
    
    
            if (svec.size() == ptr || svec[ptr].empty()) {
    
    
                break;
            }

            ret.push_back(svec[ptr]);
            ++ptr;
        }

        return ret;
    }

    vector<string> svec;
    int ptr = 1;
};

猜你喜欢

转载自blog.csdn.net/tus00000/article/details/112622418
今日推荐