[백준(BOJ)] 10828번: 스택 (스택, C++)

📄 문제풀이 과정

---

C++ STL stack 을 사용해서 스택을 구현했다.

입력 시 push에서 whitespace 뒤에 숫자가 오기 때문에  getline()으로 입력받았고, std::cin >> std::ws 으로 newline을 무시하고 엔터 시 다음 스트링을 입력받도록 했다.

switch문에서 p 문자인 경우에서 두 번째 문자로 push와 pop 구분했다.

스택이 비어있는 여부에 대해서는 삼항연산자를 사용해서 출력했다.

 

 

💻 코드

---

#include <bits/stdc++.h>
using namespace std;

int main(){
    stack<int> s;
    int N;
    scanf("%d", &N); // N: # of instructions
    
    for(int i=0; i<N; i++){
        string str;
        getline(cin >> std::ws, str);
       
        switch(str[0]){
            case 'p':
            {
                if(str[1] == 'u') s.push(stoi(str.substr(str.find(" ")+1)));
                else{
                    if(s.empty()) cout << -1 << endl;
                    else{
                        cout << s.top() << endl;
                        s.pop();
                    }	
                }
            }
            break;
            case 's':
            cout << s.size() << endl;
            break;
            case 'e':
            cout << (s.empty() ? 1 : 0) << endl;
            break;
            case 't':
            cout << (s.empty() ? -1 : s.top()) << endl;
            break;
        }
    }
    
}

 

 

🖨 제출결과

---