diff options
Diffstat (limited to 'store/works/solutions/leetcode/cpp/155-2.cpp')
| -rw-r--r-- | store/works/solutions/leetcode/cpp/155-2.cpp | 48 | 
1 files changed, 48 insertions, 0 deletions
diff --git a/store/works/solutions/leetcode/cpp/155-2.cpp b/store/works/solutions/leetcode/cpp/155-2.cpp new file mode 100644 index 0000000..aa07eee --- /dev/null +++ b/store/works/solutions/leetcode/cpp/155-2.cpp @@ -0,0 +1,48 @@ +#include <vector>
 +#include <algorithm>
 +
 +class MinStack
 +{
 +public:
 +    /** initialize your data structure here. */
 +    MinStack()
 +    {
 +        // Although I really think this is just a trick for this problem.
 +        // Leetcode does not give the input size. So I just make a reasonable assumption.
 +        data_.reserve(10000);
 +        min_stack_.reserve(10000);
 +    }
 +
 +    void push(int x)
 +    {
 +        data_.push_back(x);
 +        if (min_stack_.empty())
 +        {
 +            min_stack_.push_back(x);
 +        }
 +        else
 +        {
 +            min_stack_.push_back(std::min(min_stack_.back(), x));
 +        }
 +    }
 +
 +    void pop()
 +    {
 +        data_.pop_back();
 +        min_stack_.pop_back();
 +    }
 +
 +    int top()
 +    {
 +        return data_.back();
 +    }
 +
 +    int getMin()
 +    {
 +        return min_stack_.back();
 +    }
 +
 +private:
 +    std::vector<int> data_;
 +    std::vector<int> min_stack_;
 +};
  | 
