문제설명
입출력 예제
개념
잘못 부른 수를 0으로 지우고, 마지막의 수의 합을 출력하므로 스택의 개념을 이용하면 쉽게 풀 수 있다. 즉, 0을 입력받는 순간 스택에서 하나를 뽑으면 된다.
풀이
#include <iostream>
#include <stack>
int main()
{
int K, n, sum = 0;
std::cin >> K;
std::stack<int> st;
while (K--)
{
std::cin >> n;
if (n == 0)
st.pop();
else
st.push(n);
}
while (!st.empty())
{
sum += st.top();
st.pop();
}
std::cout << sum;
}
반응형
'Algorithm > 백준' 카테고리의 다른 글
[백준] 4949 균형잡힌 세상 with C++ (0) | 2023.05.11 |
---|---|
[백준] 9012 괄호 with C++ (0) | 2023.05.10 |
[백준] 10828 스택 with C++ (0) | 2023.05.08 |
[백준] 1010 다리 놓기 with C++ (0) | 2023.05.06 |
[백준] 20920 영단어 암기는 괴로워 with C++ (0) | 2023.05.05 |