[LeetCode] 155. Min Stack

https://leetcode.com/problems/min-stack/

Min Stack - LeetCode

Min Stack - Easy

 

 

stack을 구현하는데, 필요할때마다 stack 내의 최소값을 뽑아내는 기능을 추가해야 한다.

다른 것은 stack과 같아 stack STL을 사용하면 된다.

다만, 저장 자료형을 int가 아닌 pair<int, int> 로 하여, val과 그때의 최소값을 추가해주면 된다.

그때의 최소값은 stack이 비어있을경우에는 그 val로, 비어있지 않을 경우엔 top의 최소값과 현재 val 중 작은값으로 저장을 해주자.

 

전체 풀이 코드

 

class MinStack {
public:
    stack<pair<int, int>> st;
    MinStack() {while(!st.empty()) st.pop();}
    void push(int val) {
        if (st.empty()) st.push(make_pair(val, val));
        else (st.push(make_pair(val, min(val, st.top().second))));
    }
    void pop() {st.pop();}
    int top() {return st.top().first;} 
    int getMin() {return st.top().second;}
};