Rust每日一题(4)---数据结构-栈-minStack

  • Po
  • 更新于 2022-08-25 21:59
  • 阅读 1804

Rust每日一题(4)---数据结构-栈-minStack

Rust每日一题(4)---数据结构-栈-minStack

leetcode地址 设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。

实现 MinStack 类:

MinStack() 初始化堆栈对象。
void push(int val) 将元素val推入堆栈。
void pop() 删除堆栈顶部的元素。
int top() 获取堆栈顶部的元素。
int getMin() 获取堆栈中的最小元素。

示例 1:

输入:
["MinStack","push","push","push","getMin","pop","top","getMin"]
[[],[-2],[0],[-3],[],[],[],[]]

输出:
[null,null,null,null,-3,null,0,-2]

解释:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.getMin();   --> 返回 -2.

难度: 中等

知识点

思路

核心是每次压栈和出栈的时候需要记录最小值,有两种思路:

  1. 一个栈stack正常记录元素,另外构造一个minStack栈,每次压栈时把小于等于minStack栈顶元素的元素入栈;出栈时则只有与minStack栈顶元素相等时,minStack也弹栈;
  2. 想办法减小内存,可以只用一个栈,入栈时同时压入当前元素和最小值,出栈时也是一起出。
struct MinStack {
    stack:Vec<i32>
}

/**
 * `&self` means the method takes an immutable reference.
 * If you need a mutable reference, change it to `&mut self` instead.
 */
impl MinStack {

    fn new() -> Self {
        MinStack { stack: Vec::new()}
    }

    fn push(&mut self, val: i32) {
        if self.stack.is_empty() {
            self.stack.push(val);
            self.stack.push(val);
        } else if (val <= self.get_min()) {
            self.stack.push(val);
            self.stack.push(val);
        } else {
            self.stack.push(val);
            self.stack.push(self.top());
        }
    }

    fn pop(&mut self) {
        self.stack.pop();
        self.stack.pop();
    }

    fn top(&self) -> i32 {
        self.stack[self.stack.len() -2]
    }

    fn get_min(&self) -> i32 {
        *self.stack.last().unwrap()
    }
}
  • 原创
  • 学分: 4
  • 分类: Rust
  • 标签: Rust 
点赞 1
收藏 0
分享
本文参与登链社区写作激励计划 ,好文好收益,欢迎正在阅读的你也加入。

0 条评论

请先 登录 后评论
Po
Po
0xB332...C3ba
Blockchain & AI change the world!