【队列栈】 155. 最小栈

一时失言乱红尘 2022-12-02 15:24 115阅读 0赞

### 题目 ###

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTE3MjYwMDU_size_16_color_FFFFFF_t_70_pic_center]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTE3MjYwMDU_size_16_color_FFFFFF_t_70_pic_center 1]

### 解答 ###

#include <iostream>
    #include <string>
    #include <vector>
    #include <stack>
    #include <cassert>
    #include <list>
     
    class MinStack { 
    public:
      /** initialize your data structure here. */
      MinStack() { 
      }
    
      void push(int x) { 
        data_stack_.push(x);
    
        if (min_number_stack_.empty()) { 
          // 如果栈内还没有元素，那么第一个元素就是最小。
          min_number_stack_.push(x);
        } else { 
          int min_data = x < min_number_stack_.top() ? x : min_number_stack_.top();
          min_number_stack_.push(min_data);
        }
      }
    
      void pop() { 
        data_stack_.pop();
        min_number_stack_.pop();
      }
    
      int top() { 
        assert(!data_stack_.empty());
        return data_stack_.top();
      }
    
      int getMin() { 
        assert(!min_number_stack_.empty());
        return min_number_stack_.top();
      }
    
    private:
      // ！！！这是一种很好的设计技巧！！！想办法保存每个阶段的状态！！！
      // 由于需要知道每个位置的最小值，所以O(1)的办法就是记录每个位置的最小值。
      // 使用两个栈，一个记录数据，一个记录当前位置的最小数。
      // 用空间换取时间。
      std::stack<int> data_stack_;
      std::stack<int> min_number_stack_;
    };
    
    int main() { 
      MinStack min_stack;
      min_stack.push(-1);
      min_stack.push(-2);
      min_stack.push(3);
      min_stack.push(-10);
      min_stack.push(100);
    
      std::cout << min_stack.getMin() << std::endl;
    
      return 0;
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTE3MjYwMDU_size_16_color_FFFFFF_t_70_pic_center]: /images/20221123/5b2e58938db24993b386af322c33539e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTE3MjYwMDU_size_16_color_FFFFFF_t_70_pic_center 1]: /images/20221123/9393de4c75a64b39a4129ea7b9e2574a.png