【Rust—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”]

解题思路


    使用HashMap储存id_key、value 先将值插入到Map中 判断id_key是否与ptr相同 如果不相同直接结束,返回空 反之,返回当前value以及按 id_key 递增的值 并且ptr = id_key+1

😺代码😺

use std::collections::HashMap;
struct OrderedStream {
          
   
    ptr: i32,
    map: HashMap<i32, String>,
}

/**
 * `&self` means the method takes an immutable reference.
 * If you need a mutable reference, change it to `&mut self` instead.
 */
impl OrderedStream {
          
   
    fn new(n: i32) -> Self {
          
   
        OrderedStream {
          
   
            ptr: 1,
            map: HashMap::with_capacity(n as usize),
        }
    }

    fn insert(&mut self, id_key: i32, value: String) -> Vec<String> {
          
   
        let mut ans = Vec::new();
        let s = value.clone();

        self.map.insert(id_key, value);

        if self.ptr == id_key {
          
   
            ans.push(s);

            let mut i = 1;
            l: loop {
          
   
                match self.map.get(&(id_key + i)) {
          
   
                    Some(s) => {
          
   
                        ans.push(s.clone());
                        i += 1;
                    }
                    None => break l,
                }
            }
            self.ptr = id_key + i;
        }
        ans
    }
}

🐯执行结果🐯

🌀更多题解🌀


    更多力扣题解请到我的仓库:

经验分享 程序员 微信小程序 职场和发展