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.

难度: 中等

知识点

• 栈
• vector相关API

思路

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

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()
    }
}