10828번 - 스택
시뮬레이션 문제, 나는 vector 를 이용해서 스택을 구현했다.
stack 은 마지막에 들어온게 처음에 나가는 LIFO 구조란 점만 기억해서 문제를 풀어주면 끝!
### push 1 이런식으로 숫자를 받는 경우, cin.clear() 를 해줘야 한다는 점
<정답 코드>
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | #include<iostream> #include<string> #include<vector> using namespace std; int main() { int N; cin >> N; vector<int> st; while (N--) { string s; cin >> s; cin.clear(); if (s == "push") { int k; cin >> k; st.push_back(k); } else if (s == "top") { if (st.size() == 0) { cout << "-1\n"; } else { cout << st[st.size()-1] << "\n"; } } else if (s == "size") { cout << st.size() << "\n"; } else if (s == "empty") { if (st.size() == 0) { cout << "1\n"; } else { cout << "0\n"; } } else if (s == "pop") { if (st.size() == 0) { cout << "-1\n"; } else { cout << st[st.size()-1] << "\n"; st.pop_back(); } } } return 0; } | cs |
반응형