LeetCode 1656. 设计有序流

你的名字 2023-01-06 12:00 96阅读 0赞

有 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;
    };