스택을 활용한 간단한 구현 문제였다.
K개의 숫자를 반복문을 활용해 한개씩 받는다.
스택에 있는 수를 모두 더한다.
#include <iostream>
#include <stack>
using namespace std;
int main() {
int k;
cin >> k;
stack<int> st;
for (int i = 0; i < k; i++) {
int tmp;
cin >> tmp;
if (tmp == 0)
st.pop();
else {
st.push(tmp);
}
}
int sum = 0;
while (!st.empty()) {
sum += st.top();
st.pop();
}
cout << sum;
}