[C++] BOJ (백준) 10773 : 제로

문제

10773번: 제로 (acmicpc.net)

10773번: 제로

코드

#include <iostream>
#include <stack>

using namespace std;

int main() {
  ios_base::sync_with_stdio(false);
  cin.tie(nullptr);

  stack<int> st;
  int k;
  cin >> k;
  while (k--) {
    int tmp;
    cin >> tmp;
    if (tmp == 0) st.pop();
    else st.emplace(tmp);
  }

  int sum = 0;
  while (!st.empty()) {
    sum += st.top();
    st.pop();
  }
  cout << sum;
  return 0;
}

 

설명

0을 입력받을 때마다 가장 최근의 수를 지운다.
즉, 가장 마지막에 입력된 값을 관리해야하기 때문에 LIFO 자료구조인 스택을 사용하여야 한다.