【算法】1656. 设计有序流(多语言实现)

109 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 11 天,点击查看活动详情


1656. 设计有序流:

n(id, value) 对,其中 id1n 之间的一个整数,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
    • 否则,返回一个空列表。

样例 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

原题传送门:

leetcode.cn/problems/de…


分析

  • 面对这道算法题目,二当家的陷入了沉思。
  • 好像照着题意做就行了,唯一要考虑的是value用什么结构去存储,数组刚刚好。

题解

rust

struct OrderedStream {
    values: Vec<String>,
    ptr: usize,
}


/**
 * `&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 {
            values: vec!["".to_string(); n as usize],
            ptr: 0,
        }
    }

    fn insert(&mut self, id_key: i32, value: String) -> Vec<String> {
        self.values[(id_key - 1) as usize] = value;
        let mut res = Vec::new();
        while self.ptr < self.values.len() && self.values[self.ptr] != "" {
            res.push(self.values[self.ptr].clone());
            self.ptr += 1;
        }
        res
    }
}

/**
 * Your OrderedStream object will be instantiated and called as such:
 * let obj = OrderedStream::new(n);
 * let ret_1: Vec<String> = obj.insert(idKey, value);
 */

go

type OrderedStream struct {
    values []string
    ptr    int
}

func Constructor(n int) OrderedStream {
    return OrderedStream{
        values: make([]string, n),
        ptr:    0,
    }
}

func (this *OrderedStream) Insert(idKey int, value string) []string {
    this.values[idKey-1] = value
    start := this.ptr
    for this.ptr < len(this.values) && this.values[this.ptr] != "" {
        this.ptr++
    }
    return this.values[start:this.ptr]
}


/**
 * Your OrderedStream object will be instantiated and called as such:
 * obj := Constructor(n);
 * param_1 := obj.Insert(idKey,value);
 */

c++

class OrderedStream {
private:
    vector<string> values;
    int ptr;
public:
    OrderedStream(int n) {
        values.resize(n);
        ptr = 0;
    }

    vector<string> insert(int idKey, string value) {
        values[idKey - 1] = value;
        vector<string> res;
        while (ptr < values.size() && values[ptr] != "") {
            res.emplace_back(values[ptr]);
            ++ptr;
        }
        return res;
    }
};

/**
 * Your OrderedStream object will be instantiated and called as such:
 * OrderedStream* obj = new OrderedStream(n);
 * vector<string> param_1 = obj->insert(idKey,value);
 */

java

class OrderedStream {
    private final String[] values;
    private int ptr;

    public OrderedStream(int n) {
        values = new String[n];
        ptr = 0;
    }

    public List<String> insert(int idKey, String value) {
        values[idKey - 1] = value;
        List<String> res = new ArrayList<>();
        while (ptr < values.length && values[ptr] != null) {
            res.add(values[ptr]);
            ++ptr;
        }
        return res;
    }
}

/**
 * Your OrderedStream object will be instantiated and called as such:
 * OrderedStream obj = new OrderedStream(n);
 * List<String> param_1 = obj.insert(idKey,value);
 */

python

class OrderedStream:

    def __init__(self, n: int):
        self.values = [""] * n
        self.ptr = 0

    def insert(self, idKey: int, value: str) -> List[str]:
        self.values[idKey - 1] = value
        res = []
        while self.ptr < len(self.values) and self.values[self.ptr]:
            res.append(self.values[self.ptr])
            self.ptr += 1
        return res


# Your OrderedStream object will be instantiated and called as such:
# obj = OrderedStream(n)
# param_1 = obj.insert(idKey,value)


非常感谢你阅读本文~
放弃不难,但坚持一定很酷~
希望我们大家都能每天进步一点点~
本文由 二当家的白帽子:https://juejin.cn/user/2771185768884824/posts 博客原创~


开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 11 天,点击查看活动详情